Files
DeOldify/ImageColorizerStable.ipynb
T
2019-03-25 17:52:16 -07:00

71 KiB

In [ ]:
import os
os.environ['CUDA_VISIBLE_DEVICES']='2' 
In [ ]:
from fasterai.visualize import *
plt.style.use('dark_background')
In [ ]:
#Adjust render_factor (int) if image doesn't look quite right (max 64 on 11GB GPU).  The default here works for most photos.  
#It literally just is a number multiplied by 16 to get the square render resolution.  
#Note that this doesn't affect the resolution of the final output- the output is the same resolution as the input.
#Example:  render_factor=21 => color is rendered at 16x21 = 336x336 px.  
render_factor=35
In [ ]:
vis = get_stable_image_colorizer(render_factor=render_factor)
#vis = get_image_colorizer(root_folder=Path('data/imagenet/ILSVRC/Data/CLS-LOC/bandw'), weights_name='ColorizeNew73_gen192_09_49', render_factor=render_factor)
In [ ]:
vis.plot_transformed_image("test_images/1852GatekeepersWindsor.jpg", render_factor=45)
In [ ]:
vis.plot_transformed_image("test_images/Chief.jpg", render_factor=17)
In [ ]:
vis.plot_transformed_image("test_images/1850SchoolForGirls.jpg", render_factor=45)
In [ ]:
vis.plot_transformed_image("test_images/AtlanticCityBeach1905.jpg", render_factor=32)
In [ ]:
vis.plot_transformed_image("test_images/CottonMillWorkers1913.jpg")
In [ ]:
vis.plot_transformed_image("test_images/BrooklynNavyYardHospital.jpg")
In [ ]:
vis.plot_transformed_image("test_images/FinnishPeasant1867.jpg")
In [ ]:
vis.plot_transformed_image("test_images/AtlanticCity1905.png", render_factor=40)
In [ ]:
vis.plot_transformed_image("test_images/PushingCart.jpg", render_factor=30)
In [ ]:
vis.plot_transformed_image("test_images/Drive1905.jpg")
In [ ]:
vis.plot_transformed_image("test_images/IronLung.png", render_factor=26)
In [ ]:
vis.plot_transformed_image("test_images/FamilyWithDog.jpg")
In [ ]:
vis.plot_transformed_image("test_images/DayAtSeaBelgium.jpg")
In [ ]:
vis.plot_transformed_image("test_images/marilyn_woods.jpg", render_factor=25)
In [ ]:
vis.plot_transformed_image("test_images/OldWomanSweden1904.jpg", render_factor=45)
In [ ]:
vis.plot_transformed_image("test_images/WomenTapingPlanes.jpg")
In [ ]:
vis.plot_transformed_image("test_images/overmiller.jpg", render_factor=30)
In [ ]:
vis.plot_transformed_image("test_images/BritishDispatchRider.jpg", render_factor=16)
In [ ]:
vis.plot_transformed_image("test_images/MuseauNacionalDosCoches.jpg", render_factor=40)
In [ ]:
vis.plot_transformed_image("test_images/abe.jpg", render_factor=13)
In [ ]:
vis.plot_transformed_image("test_images/RossCorbettHouseCork.jpg", render_factor=40)
In [ ]:
vis.plot_transformed_image("test_images/HPLabelleOfficeMontreal.jpg", render_factor=45)
In [ ]:
vis.plot_transformed_image("test_images/einstein_beach.jpg", render_factor=30)
In [ ]:
vis.plot_transformed_image("test_images/airmen1943.jpg")
In [ ]:
vis.plot_transformed_image("test_images/20sWoman.jpg", render_factor=24)
In [ ]:
vis.plot_transformed_image("test_images/egypt-1.jpg", render_factor=18)
In [ ]:
vis.plot_transformed_image("test_images/Rutherford_Hayes.jpg")
In [ ]:
vis.plot_transformed_image("test_images/einstein_portrait.jpg", render_factor=20)
In [ ]:
vis.plot_transformed_image("test_images/pinkerton.jpg", render_factor=11)
In [ ]:
vis.plot_transformed_image("test_images/WaltWhitman.jpg", render_factor=12)
In [ ]:
vis.plot_transformed_image("test_images/dorothea-lange.jpg", render_factor=40)
In [ ]:
vis.plot_transformed_image("test_images/Hemmingway2.jpg", render_factor=24)
In [ ]:
vis.plot_transformed_image("test_images/hemmingway.jpg", render_factor=14)
In [ ]:
vis.plot_transformed_image("test_images/smoking_kid.jpg", render_factor=35)
In [ ]:
vis.plot_transformed_image("test_images/teddy_rubble.jpg", render_factor=42)
In [ ]:
vis.plot_transformed_image("test_images/dustbowl_2.jpg", render_factor=16)
In [ ]:
vis.plot_transformed_image("test_images/camera_man.jpg", render_factor=33)
In [ ]:
vis.plot_transformed_image("test_images/migrant_mother.jpg", render_factor=42)
In [ ]:
vis.plot_transformed_image("test_images/marktwain.jpg", render_factor=14)
In [ ]:
vis.plot_transformed_image("test_images/HelenKeller.jpg", render_factor=35)
In [ ]:
vis.plot_transformed_image("test_images/Evelyn_Nesbit.jpg", render_factor=42)
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", render_factor=28)
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", render_factor=30)
In [ ]:
vis.plot_transformed_image("test_images/last_samurai.jpg", render_factor=20)
In [ ]:
vis.plot_transformed_image("test_images/AnselAdamsWhiteChurch.jpg", render_factor=25)
In [ ]:
vis.plot_transformed_image("test_images/opium.jpg", render_factor=30)
In [ ]:
vis.plot_transformed_image("test_images/dorothea_lange_2.jpg", render_factor=42)
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", render_factor=45)
In [ ]:
vis.plot_transformed_image("test_images/ww1_trench.jpg", render_factor=18)
In [ ]:
vis.plot_transformed_image("test_images/women-bikers.png", render_factor=23)
In [ ]:
vis.plot_transformed_image("test_images/Unidentified1855.jpg", render_factor=119)
In [ ]:
vis.plot_transformed_image("test_images/skycrapper_lunch.jpg", render_factor=25)
In [ ]:
vis.plot_transformed_image("test_images/sioux.jpg", render_factor=28)
In [ ]:
vis.plot_transformed_image("test_images/school_kids.jpg", render_factor=20)
In [ ]:
vis.plot_transformed_image("test_images/royal_family.jpg", render_factor=42)
In [ ]:
vis.plot_transformed_image("test_images/redwood_lumberjacks.jpg", render_factor=45)
In [ ]:
vis.plot_transformed_image("test_images/poverty.jpg", render_factor=40)
In [ ]:
vis.plot_transformed_image("test_images/paperboy.jpg", render_factor=45)
In [ ]:
vis.plot_transformed_image("test_images/NativeAmericans.jpg", render_factor=21)
In [ ]:
vis.plot_transformed_image("test_images/helmut_newton-.jpg")
In [ ]:
vis.plot_transformed_image("test_images/Greece1911.jpg", render_factor=24)
In [ ]:
vis.plot_transformed_image("test_images/FatMenClub.jpg", render_factor=18)
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", render_factor=28)
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", render_factor=21)
In [ ]:
vis.plot_transformed_image("test_images/covered-wagons-traveling.jpg")
In [ ]:
vis.plot_transformed_image("test_images/civil-war_2.jpg", render_factor=42)
In [ ]:
vis.plot_transformed_image("test_images/civil_war_4.jpg")
In [ ]:
vis.plot_transformed_image("test_images/civil_war_3.jpg", render_factor=28)
In [ ]:
vis.plot_transformed_image("test_images/civil_war.jpg")
In [ ]:
vis.plot_transformed_image("test_images/BritishSlum.jpg", render_factor=30)
In [ ]:
vis.plot_transformed_image("test_images/bicycles.jpg", render_factor=27)
In [ ]:
vis.plot_transformed_image("test_images/brooklyn_girls_1940s.jpg")
In [ ]:
vis.plot_transformed_image("test_images/40sCouple.jpg", render_factor=21)
In [ ]:
vis.plot_transformed_image("test_images/1946Wedding.jpg")
In [ ]:
vis.plot_transformed_image("test_images/Dolores1920s.jpg", render_factor=18)
In [ ]:
vis.plot_transformed_image("test_images/TitanicGym.jpg", render_factor=38)
In [ ]:
vis.plot_transformed_image("test_images/FrenchVillage1950s.jpg", render_factor=32)
In [ ]:
vis.plot_transformed_image("test_images/ClassDivide1930sBrittain.jpg", render_factor=17)
In [ ]:
vis.plot_transformed_image("test_images/1870sSphinx.jpg")
In [ ]:
vis.plot_transformed_image("test_images/1890Surfer.png", render_factor=37)
In [ ]:
vis.plot_transformed_image("test_images/TV1930s.jpg")
Warning:
Output truncated. This notebook contains too many cells to display efficiently.