opensource/DeOldify
3
0
Fork 0
mirror of https://github.com/jantic/DeOldify.git synced 2026-11-04 06:59:56 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
33343bc2f288521cc48962a8057455ca376fa744
DeOldify/ColorizeVisualization.ipynb
T

59 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 fasterai.filters import *
from fastai.torch_imports import *
from pathlib import Path
from itertools import repeat
import tensorboardX
plt.style.use('dark_background')
torch.backends.cudnn.benchmark=True
torch.cuda.set_device(0)
In [ ]:
IMAGENET = Path('data/imagenet/ILSVRC/Data/CLS-LOC/train')
#colorizer_path = IMAGENET.parent/('colorize_gen_192.h5')
colorizer_path = IMAGENET.parent/('colorize34_gen_192.h5')
#colorizer_path = IMAGENET.parent/('colorize18_gen_160.h5')
defader_path = IMAGENET.parent/('bwdefade3_gen_160.h5')

#The higher the render_factor, the more GPU memory will be used and generally images will look better.  
#11GB can take a factor of 42 max.  Performance generally gracefully degrades with lower factors, 
#though you may also find that certain images will actually render better at lower numbers.  
#This tends to be the case with the oldest photos.
render_factor=42
In [ ]:
filters = [Colorizer34(gpu=0, weights_path=colorizer_path,nf_factor=2, map_to_orig=True)]
#filters = [DeFader(gpu=3, weights_path=defader_path, nf_factor=2), Colorizer34(gpu=0, weights_path=colorizer_path,nf_factor=2, map_to_orig=True)]
vis = ModelImageVisualizer(filters, render_factor=render_factor, results_dir='result_images')
In [ ]:
vis.plot_transformed_image("test_images/1852GatekeepersWindsor.jpg")
In [ ]:
vis.plot_transformed_image("test_images/Chief.jpg", render_factor=17)
In [ ]:
vis.plot_transformed_image("test_images/1850SchoolForGirls.jpg", render_factor=42)
In [ ]:
vis.plot_transformed_image("test_images/AtlanticCityBeach1905.jpg", render_factor=42)
In [ ]:
vis.plot_transformed_image("test_images/CottonMillWorkers1913.jpg", render_factor=41)
In [ ]:
vis.plot_transformed_image("test_images/BrooklynNavyYardHospital.jpg")
In [ ]:
vis.plot_transformed_image("test_images/FinnishPeasant1867.jpg", render_factor=20)
In [ ]:
vis.plot_transformed_image("test_images/AtlanticCity1905.png", render_factor=39)
In [ ]:
vis.plot_transformed_image("test_images/PushingCart.jpg")
In [ ]:
vis.plot_transformed_image("test_images/Drive1905.jpg")
In [ ]:
vis.plot_transformed_image("test_images/IronLung.png", render_factor=39)
In [ ]:
vis.plot_transformed_image("test_images/FamilyWithDog.jpg")
In [ ]:
vis.plot_transformed_image("test_images/DayAtSeaBelgium.jpg", render_factor=41)
In [ ]:
vis.plot_transformed_image("test_images/marilyn_woods.jpg", render_factor=30)
In [ ]:
vis.plot_transformed_image("test_images/OldWomanSweden1904.jpg", render_factor=39)
In [ ]:
vis.plot_transformed_image("test_images/WomenTapingPlanes.jpg")
In [ ]:
vis.plot_transformed_image("test_images/overmiller.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/abe.jpg", render_factor=25)
In [ ]:
vis.plot_transformed_image("test_images/RossCorbettHouseCork.jpg", render_factor=33)
In [ ]:
vis.plot_transformed_image("test_images/HPLabelleOfficeMontreal.jpg")
In [ ]:
vis.plot_transformed_image("test_images/einstein_beach.jpg", render_factor=42)
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", render_factor=36)
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/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/hemmingway.jpg")
In [ ]:
vis.plot_transformed_image("test_images/smoking_kid.jpg")
In [ ]:
vis.plot_transformed_image("test_images/teddy_rubble.jpg", render_factor=40)
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", render_factor=35)
In [ ]:
vis.plot_transformed_image("test_images/unnamed.jpg", render_factor=40)
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), render_factor=42)
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", render_factor=40)
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", render_factor=31)
In [ ]:
vis.plot_transformed_image("test_images/EgyptColosus.jpg", render_factor=31)
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=19)
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")
Warning:
Output truncated. This notebook contains too many cells to display efficiently.
Reference in New Issue View Git Blame Copy Permalink