अपने स्थानीय TensorFlow के लिए RSVP आज हर जगह घटना!
इस पेज का अनुवाद Cloud Translation API से किया गया है.
Switch to English

असीम कोलाब

TensorFlow.org पर देखें Google Colab में चलाएं GitHub पर देखें नोटबुक डाउनलोड करें TF हब मॉडल देखें

असीम मॉडल कोलाब में आपका स्वागत है! यह नोटबुक आपको छवियों पर मॉडल चलाने और परिणामों की कल्पना करने के चरणों के माध्यम से ले जाएगा।

अवलोकन

असीम छवि एक्सट्रपलेशन के लिए एक मॉडल है। यह मॉडल एक छवि लेता है, आंतरिक रूप से इसका एक पोर्टिओन ( 1/2 , 1/4 , 3/4 ) मास्क करता है और मास्क भाग को पूरा करता है। अधिक जानकारी के लिए असीम का संदर्भ लें : छवि विस्तार के लिए जनरेटिव एडवरसरी नेटवर्क या TensorFlow हब पर मॉडल प्रलेखन।

आयात और सेटअप

आधार आयात के साथ शुरू करें।

import tensorflow as tf
import tensorflow_hub as hub
from io import BytesIO
from PIL import Image as PilImage
import numpy as np
from matplotlib import pyplot as plt
from six.moves.urllib.request import urlopen

इनपुट के लिए छवि पढ़ना

छवि को लोड करने और इसे मॉडल के लिए प्रारूपित करने में मदद करने के लिए एक उपयोग विधि तैयार करें (257x257x3)। यह विधि विरूपण से बचने के लिए छवि को एक वर्ग में भी फसल देगी और आप स्थानीय छवियों या इंटरनेट से उपयोग कर सकते हैं।

0a2828ec0

दृश्य विधि

हम मॉडल द्वारा उत्पन्न नकाबपोश संस्करण और "भरे हुए" संस्करण के साथ मूल छवि को दिखाने के लिए एक विज़ुअलाइज़ेशन विधि भी बनाएंगे।

def visualize_output_comparison(img_original, img_masked, img_filled):
  plt.figure(figsize=(24,12))
  plt.subplot(131)
  plt.imshow((np.squeeze(img_original)))
  plt.title("Original", fontsize=24)
  plt.axis('off')
  plt.subplot(132)
  plt.imshow((np.squeeze(img_masked)))
  plt.title("Masked", fontsize=24)
  plt.axis('off')
  plt.subplot(133)
  plt.imshow((np.squeeze(img_filled)))
  plt.title("Generated", fontsize=24)
  plt.axis('off')
  plt.show()

एक छवि लोड हो रहा है

हम एक नमूना छवि लोड करेंगे, लेकिन अपनी खुद की छवि को कोलाब पर अपलोड करने के लिए स्वतंत्र हो गए और इसके साथ प्रयास करें। याद रखें कि मॉडल में मानव छवियों के संबंध में कुछ सीमाएं हैं।

wikimedia = "https://upload.wikimedia.org/wikipedia/commons/thumb/3/31/Nusfjord_road%2C_2010_09.jpg/800px-Nusfjord_road%2C_2010_09.jpg"
# wikimedia = "https://upload.wikimedia.org/wikipedia/commons/thumb/4/47/Beech_forest_M%C3%A1tra_in_winter.jpg/640px-Beech_forest_M%C3%A1tra_in_winter.jpg"
# wikimedia = "https://upload.wikimedia.org/wikipedia/commons/thumb/b/b2/Marmolada_Sunset.jpg/640px-Marmolada_Sunset.jpg"
# wikimedia = "https://upload.wikimedia.org/wikipedia/commons/thumb/9/9d/Aegina_sunset.jpg/640px-Aegina_sunset.jpg"

input_img = read_image(wikimedia)

TensorFlow हब से एक मॉडल का चयन करना

TensorFlow हब पर हमारे पास असीम मॉडल के 3 संस्करण हैं: हाफ, क्वार्टर और थ्री क्वार्टर। निम्नलिखित सेल में आप उनमें से किसी को भी चुन सकते हैं और अपनी छवि पर प्रयास कर सकते हैं। यदि आप एक दूसरे के साथ प्रयास करना चाहते हैं, तो बस इसे चुना और निम्नलिखित कोशिकाओं को एक्सपेक्ट करें।

मॉडल चयन

अब हमने जो मॉडल चुना है, उसे हम TensorFlow Hub से लोड करना चाहते हैं।

print("Loading model {} ({})".format(model_name, model_handle))
model = hub.load(model_handle)
Loading model Boundless Quarter (https://tfhub.dev/google/boundless/quarter/1)

करत आक्षेप

असीम मॉडल के दो आधार होते हैं:

  • एक मुखौटा के साथ इनपुट छवि लागू होती है
  • इसे पूरा करने के लिए एक्सट्रपलेशन के साथ नकाबपोश छवि

हम तुलनात्मक दृश्य दिखाने के लिए इन दो चित्रों का उपयोग कर सकते हैं।

result = model.signatures['default'](tf.constant(input_img))
generated_image =  result['default']
masked_image = result['masked_image']

visualize_output_comparison(input_img, masked_image, generated_image)

पींग