DeOldify/ColorizeVisualization.ipynb at dabb3a00edb7300a0f71cf97df6c2a9bda184799

mirror of https://github.com/jantic/DeOldify.git synced 2026-11-03 22:44:30 +08:00

Files

T

Jason Antic dabb3a00ed Initial commit of new filter functionality (big!)

Includes chrominance/luminance optimzation, change to using render_factor instead of size for easier "optimal result image" search;  changing to rendering in model using stretched full squares for real world images.  Al these changes amount to huge bump in default quality, more flexibility in hardware requirements, and graceful degradation in rendering as memory is descreased.

2018-11-14 21:32:01 -08:00

55 KiB

Raw Blame History

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)
torch.no_grad()
plt.style.use('dark_background')
torch.backends.cudnn.benchmark=True

In [ ]:

IMAGENET = Path('data/imagenet/ILSVRC/Data/CLS-LOC/train')
colorizer_path = IMAGENET.parent/('colorize_gen_192.h5')

#Can be bigger for GPUs with more memory.  11GB can take a factor of 21 max.  Performance generally gracefully degrades with lower factors.
render_factor=21

In [ ]:

filters = [Colorizer(gpu=0, weights_path=colorizer_path)]
vis = ModelImageVisualizer(filters, render_factor=render_factor, results_dir='result_images')

In [ ]:

vis.plot_transformed_image("test_images/WomenTapingPlanes.jpg")

In [ ]:

vis.plot_transformed_image("test_images/BritishDispatchRider.jpg")

In [ ]:

vis.plot_transformed_image("test_images/MuseauNacionalDosCoches.jpg")

In [ ]:

vis.plot_transformed_image("test_images/RossCorbettHouseCork.jpg", render_factor=17)

In [ ]:

vis.plot_transformed_image("test_images/HPLabelleOfficeMontreal.jpg")

In [ ]:

vis.plot_transformed_image("test_images/overmiller.jpg")

In [ ]:

vis.plot_transformed_image("test_images/einstein_beach.jpg")

In [ ]:

vis.plot_transformed_image("test_images/abe.jpg", render_factor=12)

In [ ]:

vis.plot_transformed_image("test_images/airmen1943.jpg")

In [ ]:

vis.plot_transformed_image("test_images/20sWoman.jpg")

In [ ]:

vis.plot_transformed_image("test_images/egypt-1.jpg")

In [ ]:

vis.plot_transformed_image("test_images/Rutherford_Hayes.jpg")

In [ ]:

vis.plot_transformed_image("test_images/einstein_portrait.jpg")

In [ ]:

vis.plot_transformed_image("test_images/pinkerton.jpg")

In [ ]:

vis.plot_transformed_image("test_images/marilyn_woods.jpg")

In [ ]:

vis.plot_transformed_image("test_images/WaltWhitman.jpg")

In [ ]:

vis.plot_transformed_image("test_images/dorothea-lange.jpg")

In [ ]:

vis.plot_transformed_image("test_images/Hemmingway2.jpg")

In [ ]:

vis.plot_transformed_image("test_images/Chief.jpg")

In [ ]:

vis.plot_transformed_image("test_images/hemmingway.jpg")

In [ ]:

vis.plot_transformed_image("test_images/smoking_kid.jpg")

In [ ]:

vis.plot_transformed_image("test_images/teddy_rubble.jpg")

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_2.jpg")

In [ ]:

vis.plot_transformed_image("test_images/camera_man.jpg")

In [ ]:

vis.plot_transformed_image("test_images/migrant_mother.jpg")

In [ ]:

vis.plot_transformed_image("test_images/marktwain.jpg")

In [ ]:

vis.plot_transformed_image("test_images/HelenKeller.jpg")

In [ ]:

vis.plot_transformed_image("test_images/Evelyn_Nesbit.jpg")

In [ ]:

vis.plot_transformed_image("test_images/Eddie-Adams.jpg")

In [ ]:

vis.plot_transformed_image("test_images/soldier_kids.jpg")

In [ ]:

vis.plot_transformed_image("test_images/AnselAdamsYosemite.jpg")

In [ ]:

vis.plot_transformed_image("test_images/unnamed.jpg")

In [ ]:

vis.plot_transformed_image("test_images/workers_canyon.jpg")

In [ ]:

vis.plot_transformed_image("test_images/CottonMill.jpg")

In [ ]:

vis.plot_transformed_image("test_images/JudyGarland.jpeg")

In [ ]:

vis.plot_transformed_image("test_images/kids_pit.jpg")

In [ ]:

vis.plot_transformed_image("test_images/last_samurai.jpg")

In [ ]:

vis.plot_transformed_image("test_images/AnselAdamsWhiteChurch.jpg")

In [ ]:

vis.plot_transformed_image("test_images/opium.jpg")

In [ ]:

vis.plot_transformed_image("test_images/dorothea_lange_2.jpg")

In [ ]:

vis.plot_transformed_image("test_images/rgs.jpg")

In [ ]:

vis.plot_transformed_image("test_images/wh-auden.jpg")

In [ ]:

vis.plot_transformed_image("test_images/w-b-yeats.jpg")

In [ ]:

vis.plot_transformed_image("test_images/marilyn_portrait.jpg")

In [ ]:

vis.plot_transformed_image("test_images/wilson-slaverevivalmeeting.jpg")

In [ ]:

vis.plot_transformed_image("test_images/ww1_trench.jpg")

In [ ]:

vis.plot_transformed_image("test_images/women-bikers.png", figsize=(60,60))

In [ ]:

vis.plot_transformed_image("test_images/Unidentified1855.jpg")

In [ ]:

vis.plot_transformed_image("test_images/skycrapper_lunch.jpg")

In [ ]:

vis.plot_transformed_image("test_images/sioux.jpg")

In [ ]:

vis.plot_transformed_image("test_images/school_kids.jpg")

In [ ]:

vis.plot_transformed_image("test_images/royal_family.jpg")

In [ ]:

vis.plot_transformed_image("test_images/redwood_lumberjacks.jpg")

In [ ]:

vis.plot_transformed_image("test_images/poverty.jpg")

In [ ]:

vis.plot_transformed_image("test_images/paperboy.jpg")

In [ ]:

vis.plot_transformed_image("test_images/NativeAmericans.jpg")

In [ ]:

vis.plot_transformed_image("test_images/helmut_newton-.jpg")

In [ ]:

vis.plot_transformed_image("test_images/Greece1911.jpg")

In [ ]:

vis.plot_transformed_image("test_images/FatMenClub.jpg")

In [ ]:

vis.plot_transformed_image("test_images/EgyptColosus.jpg")

In [ ]:

vis.plot_transformed_image("test_images/egypt-2.jpg")

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_sd.jpg")

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_people.jpg")

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_5.jpg")

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_1.jpg")

In [ ]:

vis.plot_transformed_image("test_images/DriveThroughGiantTree.jpg")

In [ ]:

vis.plot_transformed_image("test_images/covered-wagons-traveling.jpg", render_factor=14)

In [ ]:

vis.plot_transformed_image("test_images/civil-war_2.jpg")

In [ ]:

vis.plot_transformed_image("test_images/civil_war_4.jpg")

In [ ]:

vis.plot_transformed_image("test_images/civil_war_3.jpg")

In [ ]:

vis.plot_transformed_image("test_images/civil_war.jpg")

In [ ]:

vis.plot_transformed_image("test_images/BritishSlum.jpg")

In [ ]:

vis.plot_transformed_image("test_images/bicycles.jpg")

In [ ]:

vis.plot_transformed_image("test_images/brooklyn_girls_1940s.jpg")

In [ ]:

vis.plot_transformed_image("test_images/40sCouple.jpg")

In [ ]:

vis.plot_transformed_image("test_images/1946Wedding.jpg")

In [ ]:

vis.plot_transformed_image("test_images/Dolores1920s.jpg")

In [ ]:

vis.plot_transformed_image("test_images/TitanicGym.jpg")

In [ ]:

vis.plot_transformed_image("test_images/FrenchVillage1950s.jpg")

In [ ]:

vis.plot_transformed_image("test_images/ClassDivide1930sBrittain.jpg")

In [ ]:

vis.plot_transformed_image("test_images/1870sSphinx.jpg")

In [ ]:

vis.plot_transformed_image("test_images/1890Surfer.png")

In [ ]:

vis.plot_transformed_image("test_images/TV1930s.jpg")

In [ ]:

vis.plot_transformed_image("test_images/1864UnionSoldier.jpg")

In [ ]:

vis.plot_transformed_image("test_images/1890sMedStudents.png")

In [ ]:

vis.plot_transformed_image("test_images/BellyLaughWWI.jpg")

In [ ]:

vis.plot_transformed_image("test_images/PiggyBackRide.jpg")

In [ ]:

vis.plot_transformed_image("test_images/HealingTree.jpg")

In [ ]:

vis.plot_transformed_image("test_images/ManPile.jpg")

In [ ]:

vis.plot_transformed_image("test_images/1910Bike.jpg")

In [ ]:

vis.plot_transformed_image("test_images/FreeportIL.jpg")

In [ ]:

vis.plot_transformed_image("test_images/DutchBabyCoupleEllis.jpg")

In [ ]:

vis.plot_transformed_image("test_images/InuitWoman1903.png")

In [ ]:

vis.plot_transformed_image("test_images/1920sDancing.jpg")

In [ ]:

vis.plot_transformed_image("test_images/AirmanDad.jpg")

In [ ]:

vis.plot_transformed_image("test_images/1910Racket.png")

Warning:

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

55 KiB Raw Blame History

55 KiB

Raw Blame History