BigGAN দিয়ে ছবি তৈরি করা

TensorFlow.org এ দেখুন Google Colab-এ চালান GitHub এ দেখুন নোটবুক ডাউনলোড করুন TF হাব মডেল দেখুন

এই নোটবুক BigGAN ইমেজ উপলব্ধ জেনারেটর জন্য একটি ডেমো হয় মেমরি হাব ।

দেখুন arXiv উপর BigGAN কাগজ এই মডেল সম্পর্কে আরও তথ্যের জন্য [1]।

রানটাইমের সাথে সংযোগ করার পরে, এই নির্দেশাবলী অনুসরণ করে শুরু করুন:

  1. (ঐচ্ছিক) আপডেট নির্বাচিত module_path নীচের প্রথম কোড কক্ষে একটি ভিন্ন চিত্র রেজল্যুশন জন্য একটি BigGAN জেনারেটরের লোড করা হয়নি।
  2. রানটাইম> অনুক্রমে প্রতিটি কক্ষে চালনা করার জন্য চালান ক্লিক করুন।
    • পরে, যখন আপনি স্লাইডার এবং ড্রপডাউন মেনু ব্যবহার করে সেটিংস পরিবর্তন করেন তখন ইন্টারেক্টিভ ভিজ্যুয়ালাইজেশনগুলি স্বয়ংক্রিয়ভাবে আপডেট হওয়া উচিত।
    • যদি না হয়, ম্যানুয়ালি আউটপুট পুনরায় রেন্ডার সেল দ্বারা খেলা বোতামটি টিপুন।

[১] অ্যান্ড্রু ব্রক, জেফ ডোনাহু এবং কারেন সিমোনিয়ান। উচ্চ সততা প্রাকৃতিক চিত্র সংশ্লেষের জন্য বড় স্কেল GAN প্রশিক্ষণ । arXiv: 1809,11096 2018।

প্রথমে, মডিউল পাথ সেট করুন। ডিফল্টভাবে, আমরা থেকে 256x256 ইমেজ জন্য BigGAN-গভীর জেনারেটরের লোড <a href="https://tfhub.dev/deepmind/biggan-deep-256/1">https://tfhub.dev/deepmind/biggan-deep-256/1</a> । 128x128 বা 512x512 চিত্রগুলি জেনারেট করতে বা আউট সক্রিয় মূল BigGAN জেনারেটর, মন্তব্য ব্যবহার করতে module_path সেটিং এবং অন্যদের uncomment করুন।

# BigGAN-deep models
# module_path = 'https://tfhub.dev/deepmind/biggan-deep-128/1'  # 128x128 BigGAN-deep
module_path = 'https://tfhub.dev/deepmind/biggan-deep-256/1'  # 256x256 BigGAN-deep
# module_path = 'https://tfhub.dev/deepmind/biggan-deep-512/1'  # 512x512 BigGAN-deep

# BigGAN (original) models
# module_path = 'https://tfhub.dev/deepmind/biggan-128/2'  # 128x128 BigGAN
# module_path = 'https://tfhub.dev/deepmind/biggan-256/2'  # 256x256 BigGAN
# module_path = 'https://tfhub.dev/deepmind/biggan-512/2'  # 512x512 BigGAN

সেটআপ

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

import os
import io
import IPython.display
import numpy as np
import PIL.Image
from scipy.stats import truncnorm
import tensorflow_hub as hub

WARNING:tensorflow:From /tmpfs/src/tf_docs_env/lib/python3.7/site-packages/tensorflow/python/compat/v2_compat.py:111: disable_resource_variables (from tensorflow.python.ops.variable_scope) is deprecated and will be removed in a future version.
Instructions for updating:
non-resource variables are not supported in the long term

TF হাব থেকে একটি BigGAN জেনারেটর মডিউল লোড করুন

tf.reset_default_graph()
print('Loading BigGAN module from:', module_path)
module = hub.Module(module_path)
inputs = {k: tf.placeholder(v.dtype, v.get_shape().as_list(), k)
          for k, v in module.get_input_info_dict().items()}
output = module(inputs)

print()
print('Inputs:\n', '\n'.join(
    '  {}: {}'.format(*kv) for kv in inputs.items()))
print()
print('Output:', output)

Loading BigGAN module from: https://tfhub.dev/deepmind/biggan-deep-256/1
INFO:tensorflow:Saver not created because there are no variables in the graph to restore
INFO:tensorflow:Saver not created because there are no variables in the graph to restore
Inputs:
   y: Tensor("y:0", shape=(?, 1000), dtype=float32)
  z: Tensor("z:0", shape=(?, 128), dtype=float32)
  truncation: Tensor("truncation:0", shape=(), dtype=float32)

Output: Tensor("module_apply_default/G_trunc_output:0", shape=(?, 256, 256, 3), dtype=float32)

BigGAN চিত্রের নমুনা এবং প্রদর্শনের জন্য কিছু ফাংশন সংজ্ঞায়িত করুন

input_z = inputs['z']
input_y = inputs['y']
input_trunc = inputs['truncation']

dim_z = input_z.shape.as_list()[1]
vocab_size = input_y.shape.as_list()[1]

def truncated_z_sample(batch_size, truncation=1., seed=None):
  state = None if seed is None else np.random.RandomState(seed)
  values = truncnorm.rvs(-2, 2, size=(batch_size, dim_z), random_state=state)
  return truncation * values

def one_hot(index, vocab_size=vocab_size):
  index = np.asarray(index)
  if len(index.shape) == 0:
    index = np.asarray([index])
  assert len(index.shape) == 1
  num = index.shape[0]
  output = np.zeros((num, vocab_size), dtype=np.float32)
  output[np.arange(num), index] = 1
  return output

def one_hot_if_needed(label, vocab_size=vocab_size):
  label = np.asarray(label)
  if len(label.shape) <= 1:
    label = one_hot(label, vocab_size)
  assert len(label.shape) == 2
  return label

def sample(sess, noise, label, truncation=1., batch_size=8,
           vocab_size=vocab_size):
  noise = np.asarray(noise)
  label = np.asarray(label)
  num = noise.shape[0]
  if len(label.shape) == 0:
    label = np.asarray([label] * num)
  if label.shape[0] != num:
    raise ValueError('Got # noise samples ({}) != # label samples ({})'
                     .format(noise.shape[0], label.shape[0]))
  label = one_hot_if_needed(label, vocab_size)
  ims = []
  for batch_start in range(0, num, batch_size):
    s = slice(batch_start, min(num, batch_start + batch_size))
    feed_dict = {input_z: noise[s], input_y: label[s], input_trunc: truncation}
    ims.append(sess.run(output, feed_dict=feed_dict))
  ims = np.concatenate(ims, axis=0)
  assert ims.shape[0] == num
  ims = np.clip(((ims + 1) / 2.0) * 256, 0, 255)
  ims = np.uint8(ims)
  return ims

def interpolate(A, B, num_interps):
  if A.shape != B.shape:
    raise ValueError('A and B must have the same shape to interpolate.')
  alphas = np.linspace(0, 1, num_interps)
  return np.array([(1-a)*A + a*B for a in alphas])

def imgrid(imarray, cols=5, pad=1):
  if imarray.dtype != np.uint8:
    raise ValueError('imgrid input imarray must be uint8')
  pad = int(pad)
  assert pad >= 0
  cols = int(cols)
  assert cols >= 1
  N, H, W, C = imarray.shape
  rows = N // cols + int(N % cols != 0)
  batch_pad = rows * cols - N
  assert batch_pad >= 0
  post_pad = [batch_pad, pad, pad, 0]
  pad_arg = [[0, p] for p in post_pad]
  imarray = np.pad(imarray, pad_arg, 'constant', constant_values=255)
  H += pad
  W += pad
  grid = (imarray
          .reshape(rows, cols, H, W, C)
          .transpose(0, 2, 1, 3, 4)
          .reshape(rows*H, cols*W, C))
  if pad:
    grid = grid[:-pad, :-pad]
  return grid

def imshow(a, format='png', jpeg_fallback=True):
  a = np.asarray(a, dtype=np.uint8)
  data = io.BytesIO()
  PIL.Image.fromarray(a).save(data, format)
  im_data = data.getvalue()
  try:
    disp = IPython.display.display(IPython.display.Image(im_data))
  except IOError:
    if jpeg_fallback and format != 'jpeg':
      print(('Warning: image was too large to display in format "{}"; '
             'trying jpeg instead.').format(format))
      return imshow(a, format='jpeg')
    else:
      raise
  return disp

একটি টেনসরফ্লো সেশন তৈরি করুন এবং ভেরিয়েবল শুরু করুন

initializer = tf.global_variables_initializer()
sess = tf.Session()
sess.run(initializer)

একটি নির্দিষ্ট বিভাগের BigGAN নমুনাগুলি অন্বেষণ করুন

নানারকম চেষ্টা truncation মান।

(কোড দেখতে ঘরে ডাবল ক্লিক করুন।)

বিভাগ-শর্তাধীন নমুনা

png

BigGAN নমুনার মধ্যে ইন্টারপোলেট করুন

বিভিন্ন সেটিং ব্যবহার করে দেখুন category একই সঙ্গে গুলি noise_seed এস, অথবা একই category বিভিন্ন সঙ্গে গুলি noise_seed গুলি। অথবা বন্য যান এবং আপনি যে কোনো উপায় উভয় সেট!

(কোড দেখতে ঘরে ডাবল ক্লিক করুন।)

ইন্টারপোলেশন

png