DeOldify/ImageColorizerStableTests.ipynb at 83dc33b8ccae94ea46f69cfd498f9acedbbc3a9a

opensource/DeOldify

Fork 0

mirror of https://github.com/jantic/DeOldify.git synced 2026-10-31 03:14:33 +08:00

Files

T

Jason Antic edac73edf1 Adding warnings filter for inference on data sets being empty to avoid user confusion.

2020-07-09 14:32:32 -07:00

77 KiB

Raw Blame History

In [ ]:

#NOTE:  This must be the first call in order to work properly!
from deoldify import device
from deoldify.device_id import DeviceId
#choices:  CPU, GPU0...GPU7
device.set(device=DeviceId.GPU0)

In [ ]:

from deoldify.visualize import *
plt.style.use('dark_background')
import warnings
warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")

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_image_colorizer(render_factor=render_factor, artistic=False)
#vis = get_video_colorizer(render_factor=render_factor).vis

In [ ]:

vis.plot_transformed_image("test_images/poolparty.jpg", render_factor=45, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/1852GatekeepersWindsor.jpg", render_factor=44, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/Chief.jpg", render_factor=10, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/1850SchoolForGirls.jpg", render_factor=42, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/AtlanticCityBeach1905.jpg", render_factor=32, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/CottonMillWorkers1913.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/BrooklynNavyYardHospital.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/FinnishPeasant1867.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/AtlanticCity1905.png", render_factor=40, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/PushingCart.jpg", render_factor=24, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/Drive1905.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/IronLung.png", render_factor=26, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/FamilyWithDog.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/DayAtSeaBelgium.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/marilyn_woods.jpg", render_factor=16, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/OldWomanSweden1904.jpg", render_factor=20, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/WomenTapingPlanes.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/overmiller.jpg", render_factor=30, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/BritishDispatchRider.jpg", render_factor=16, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/MuseauNacionalDosCoches.jpg", render_factor=19, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/abe.jpg", render_factor=13, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/RossCorbettHouseCork.jpg", render_factor=40, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/HPLabelleOfficeMontreal.jpg", render_factor=44, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/einstein_beach.jpg", render_factor=32, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/airmen1943.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/20sWoman.jpg", render_factor=24, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/egypt-1.jpg", render_factor=18, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/Rutherford_Hayes.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/einstein_portrait.jpg", render_factor=15, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/pinkerton.jpg", render_factor=7, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/WaltWhitman.jpg", render_factor=9, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/dorothea-lange.jpg", render_factor=18, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/Hemmingway2.jpg", render_factor=22, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/hemmingway.jpg", render_factor=14, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/smoking_kid.jpg", render_factor=35, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/teddy_rubble.jpg", render_factor=42, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_2.jpg", render_factor=16, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/camera_man.jpg", render_factor=25, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/migrant_mother.jpg", render_factor=32, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/marktwain.jpg", render_factor=14, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/HelenKeller.jpg", render_factor=35, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/Evelyn_Nesbit.jpg", render_factor=25, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/Eddie-Adams.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/soldier_kids.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/AnselAdamsYosemite.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/unnamed.jpg", render_factor=28, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/workers_canyon.jpg", render_factor=45, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/CottonMill.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/JudyGarland.jpeg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/kids_pit.jpg", render_factor=30, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/last_samurai.jpg", render_factor=22, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/AnselAdamsWhiteChurch.jpg", render_factor=25, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/opium.jpg", render_factor=30, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/dorothea_lange_2.jpg", render_factor=42, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/rgs.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/wh-auden.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/w-b-yeats.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/marilyn_portrait.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/wilson-slaverevivalmeeting.jpg", render_factor=45, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/ww1_trench.jpg", render_factor=18, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/women-bikers.png", render_factor=23, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/Unidentified1855.jpg", render_factor=19, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/skycrapper_lunch.jpg", render_factor=25, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/sioux.jpg", render_factor=28, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/school_kids.jpg", render_factor=20, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/royal_family.jpg", render_factor=42, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/redwood_lumberjacks.jpg", render_factor=45, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/poverty.jpg", render_factor=40, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/paperboy.jpg", render_factor=45, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/NativeAmericans.jpg", render_factor=21, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/helmut_newton-.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/Greece1911.jpg", render_factor=44, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/FatMenClub.jpg", render_factor=18, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/EgyptColosus.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/egypt-2.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_sd.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_people.jpg", render_factor=24, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_5.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/dustbowl_1.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/DriveThroughGiantTree.jpg", render_factor=21, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/covered-wagons-traveling.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/civil-war_2.jpg", render_factor=42, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/civil_war_4.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/civil_war_3.jpg", render_factor=28, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/civil_war.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/BritishSlum.jpg", render_factor=30, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/bicycles.jpg", render_factor=27, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/brooklyn_girls_1940s.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/40sCouple.jpg", render_factor=21, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/1946Wedding.jpg", compare=True)

In [ ]:

vis.plot_transformed_image("test_images/Dolores1920s.jpg", render_factor=18, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/TitanicGym.jpg", render_factor=26, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/FrenchVillage1950s.jpg", render_factor=41, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/FrenchVillage1950s.jpg", render_factor=32, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/ClassDivide1930sBrittain.jpg", render_factor=45, compare=True)

In [ ]:

vis.plot_transformed_image("test_images/1870sSphinx.jpg", compare=True)

Warning:

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

77 KiB Raw Blame History

77 KiB

Raw Blame History