mirror of
https://github.com/jantic/DeOldify.git
synced 2026-11-03 22:44:30 +08:00
71 KiB
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=35In [ ]:
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.