DeOldify/ColorizeVisualization.ipynb

In [ ]:

%matplotlib inline
%reload_ext autoreload
%autoreload 2

In [ ]:

import multiprocessing
import os
from torch import autograd
from fastai.transforms import TfmType
from fasterai.transforms import *
from fastai.conv_learner import *
from fasterai.images import *
from fasterai.dataset import *
from fasterai.visualize import *
from fasterai.callbacks import *
from fasterai.loss import *
from fasterai.modules import *
from fasterai.training import *
from fasterai.generators import *
from fastai.torch_imports import *
from pathlib import Path
from itertools import repeat
import tensorboardX
torch.cuda.set_device(0)
plt.style.use('dark_background')
torch.backends.cudnn.benchmark=True

In [ ]:

IMAGENET = Path('data/imagenet/ILSVRC/Data/CLS-LOC/train')
IMAGENET_SMALL = IMAGENET/'n01440764'

colorizer_path = IMAGENET.parent/('bwc_final_gen_64.h5')

default_sz=400
torch.backends.cudnn.benchmark=True

In [ ]:

netG = Unet34(nf_factor=2).cuda()
load_model(netG, colorizer_path)
netG = netG.eval()

In [ ]:

x_tfms = [BlackAndWhiteTransform()]
data_loader = ImageGenDataLoader(sz=256, bs=8, path=IMAGENET_SMALL, random_seed=42, x_noise=False,
            keep_pct=1.0, x_tfms=x_tfms)
md = data_loader.get_model_data()

In [ ]:

vis = ModelImageVisualizer(default_sz=default_sz)

In [ ]:

vis.plot_transformed_image("test_images/overmiller.jpg", netG, md.val_ds, tfms=x_tfms, sz=420)

In [ ]:

vis.plot_transformed_image("test_images/einstein_beach.jpg", netG, md.val_ds, tfms=x_tfms,sz=600)

In [ ]:

vis.plot_transformed_image("test_images/abe.jpg", netG, md.val_ds, tfms=x_tfms, sz=480)

In [ ]:

vis.plot_transformed_image("test_images/airmen1943.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/20sWoman.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/egypt-1.jpg", netG, md.val_ds,sz=480)

In [ ]:

vis.plot_transformed_image("test_images/Rutherford_Hayes.jpg", netG, md.val_ds, tfms=x_tfms, sz=400)

In [ ]:

vis.plot_transformed_image("test_images/einstein_portrait.jpg", netG, md.val_ds, tfms=x_tfms, sz=400)

In [ ]:

vis.plot_transformed_image("test_images/pinkerton.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/marilyn_woods.jpg", netG, md.val_ds, tfms=x_tfms, sz=440)

In [ ]:

vis.plot_transformed_image("test_images/WaltWhitman.jpg", netG, md.val_ds, tfms=x_tfms, sz=280)

In [ ]:

vis.plot_transformed_image("test_images/dorothea-lange.jpg", netG, md.val_ds, tfms=x_tfms,sz=460)

In [ ]:

vis.plot_transformed_image("test_images/Hemmingway2.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/Chief.jpg", netG, md.val_ds, tfms=x_tfms,sz=520)

In [ ]:

vis.plot_transformed_image("test_images/hemmingway.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/smoking_kid.jpg", netG, md.val_ds, tfms=x_tfms,sz=550)

In [ ]:

vis.plot_transformed_image("test_images/teddy_rubble.jpg", netG, md.val_ds, tfms=x_tfms, sz=360)

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_2.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/camera_man.jpg", netG, md.val_ds, tfms=x_tfms,sz=520)

In [ ]:

vis.plot_transformed_image("test_images/migrant_mother.jpg", netG, md.val_ds, tfms=x_tfms,sz=590)

In [ ]:

vis.plot_transformed_image("test_images/marktwain.jpg", netG, md.val_ds, tfms=x_tfms, sz=530)

In [ ]:

vis.plot_transformed_image("test_images/HelenKeller.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/Evelyn_Nesbit.jpg", netG, md.val_ds, tfms=x_tfms, sz=580)

In [ ]:

vis.plot_transformed_image("test_images/Eddie-Adams.jpg", netG, md.val_ds, tfms=x_tfms, sz=460)

In [ ]:

vis.plot_transformed_image("test_images/soldier_kids.jpg", netG, md.val_ds, tfms=x_tfms, sz=400)

In [ ]:

vis.plot_transformed_image("test_images/AnselAdamsYosemite.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/unnamed.jpg", netG, md.val_ds, tfms=x_tfms, sz=550)

In [ ]:

vis.plot_transformed_image("test_images/workers_canyon.jpg", netG, md.val_ds, tfms=x_tfms,sz=570)

In [ ]:

vis.plot_transformed_image("test_images/CottonMill.jpg", netG, md.val_ds, tfms=x_tfms,sz=500)

In [ ]:

vis.plot_transformed_image("test_images/JudyGarland.jpeg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/kids_pit.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/last_samurai.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/AnselAdamsWhiteChurch.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/opium.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/dorothea_lange_2.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/rgs.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/wh-auden.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/w-b-yeats.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/marilyn_portrait.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/wilson-slaverevivalmeeting.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/ww1_trench.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/women-bikers.png", netG, md.val_ds, tfms=x_tfms, sz=450)

In [ ]:

vis.plot_transformed_image("test_images/Unidentified1855.jpg", netG, md.val_ds, tfms=x_tfms, sz=400)

In [ ]:

vis.plot_transformed_image("test_images/skycrapper_lunch.jpg", netG, md.val_ds, tfms=x_tfms, sz=550)

In [ ]:

vis.plot_transformed_image("test_images/sioux.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/school_kids.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/royal_family.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/redwood_lumberjacks.jpg", netG, md.val_ds, tfms=x_tfms, sz=550)

In [ ]:

vis.plot_transformed_image("test_images/poverty.jpg", netG, md.val_ds, tfms=x_tfms,sz=550)

In [ ]:

vis.plot_transformed_image("test_images/paperboy.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/NativeAmericans.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/helmut_newton-.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/Greece1911.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/FatMenClub.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/EgyptColosus.jpg", netG, md.val_ds, tfms=x_tfms, sz=400)

In [ ]:

vis.plot_transformed_image("test_images/egypt-2.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_sd.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_people.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_5.jpg", netG, md.val_ds, tfms=x_tfms, sz=450)

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_1.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/DriveThroughGiantTree.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/covered-wagons-traveling.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/civil-war_2.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/civil_war_4.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/civil_war_3.jpg", netG, md.val_ds, tfms=x_tfms, sz=550)

In [ ]:

vis.plot_transformed_image("test_images/civil_war.jpg", netG, md.val_ds, tfms=x_tfms, sz=540)

In [ ]:

vis.plot_transformed_image("test_images/BritishSlum.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/bicycles.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/brooklyn_girls_1940s.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/40sCouple.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/1946Wedding.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/Dolores1920s.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/TitanicGym.jpg", netG, md.val_ds, tfms=x_tfms, sz=550)

In [ ]:

vis.plot_transformed_image("test_images/FrenchVillage1950s.jpg", netG, md.val_ds, tfms=x_tfms, sz=440)

In [ ]:

vis.plot_transformed_image("test_images/ClassDivide1930sBrittain.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/1870sSphinx.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/1890Surfer.png", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/TV1930s.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/1864UnionSoldier.jpg", netG, md.val_ds, tfms=x_tfms, sz=510)

In [ ]:

vis.plot_transformed_image("test_images/1890sMedStudents.png", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/BellyLaughWWI.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/PiggyBackRide.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/HealingTree.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/ManPile.jpg", netG, md.val_ds, tfms=x_tfms, sz=450)

In [ ]:

vis.plot_transformed_image("test_images/1910Bike.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/FreeportIL.jpg", netG, md.val_ds, tfms=x_tfms, sz=402)

In [ ]:

vis.plot_transformed_image("test_images/DutchBabyCoupleEllis.jpg", netG, md.val_ds, tfms=x_tfms, sz=500)

In [ ]:

vis.plot_transformed_image("test_images/InuitWoman1903.png", netG, md.val_ds, tfms=x_tfms, sz=460)

In [ ]:

vis.plot_transformed_image("test_images/1920sDancing.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/AirmanDad.jpg", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/1910Racket.png", netG, md.val_ds, tfms=x_tfms, sz=540)

In [ ]:

vis.plot_transformed_image("test_images/1880Paris.jpg", netG, md.val_ds, tfms=x_tfms, sz=520)

In [ ]:

vis.plot_transformed_image("test_images/Deadwood1860s.png", netG, md.val_ds, tfms=x_tfms)

In [ ]:

vis.plot_transformed_image("test_images/1860sSamauris.png", netG, md.val_ds, tfms=x_tfms)

Warning:

Output truncated. This notebook contains too many cells to display efficiently.