Skip to content

Update digital Pokémon models, add compact versions of NES/Genesis mod… #490

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
joeyballentine merged 1 commit into from
Sep 18, 2025
Merged

Update digital Pokémon models, add compact versions of NES/Genesis mod… #490

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
68 changes: 68 additions & 0 deletions data/models/1x-NES-Composite-2-RGB-Small.json
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
{
"name": "NES-Composite-2-RGB-Small",
"author": "pokepress",
"license": "MIT",
"tags": [
"debanding",
"denoise",
"game-screenshots",
"restoration",
"video-frame"
],
"description": "Takes composite/RF/VHS NES footage and attempts to restore it to RGB quality. Assumes footage has been properly deinterlaced via field duplication from 240p to 480p/720p/etc. Note that:\n\n* All footage was captured in 240p/480p/720p NTSC.\n* RGB footage was captured via an AV Famicom with the RGB Blaster via the Retrotink 2x or GBS Control.\n* The model was trained exclusively on individual frames, so it can't fix things like dropouts.\n* The even and odd fields of NES composite tend to be a bit...different from each other, so there will be some jitter at 60fps.\n* I don't have access to an NES Toploader, so I wouldn't expect it to fix the jailbars very well.\n\nCompared to the OmniSR variant of this model, this version runs more quickly (despite the larger model file size), and should do a good job on footage captured directly from the console using composite. If your footage is from a VHS tape or was captured over RF, the slower OmniSR model is probably a better choice.\n\nRevision History:\n\n* 1.5.0 (09/17/2025): Added \"Small\" version of model",
"date": "2025-09-17",
"architecture": "compact",
"size": null,
"scale": 1,
"inputChannels": 3,
"outputChannels": 3,
"resources": [
{
"platform": "pytorch",
"type": "pth",
"size": 8025476,
"sha256": "6ce8754f660dd44c7c98e5c15e5125eaeb7a593d003c29b67fbf351099fcae73",
"urls": [
"https://www.mediafire.com/file/xuzy50qsayrkpxq/NES_Composite_To_RGB_compact_1_5_0.pth/file"
]
}
],
"trainingIterations": 236330,
"trainingBatchSize": 8,
"trainingHRSize": 128,
"trainingOTF": false,
"dataset": "Screenshots of NES games",
"datasetSize": 198,
"images": [
{
"type": "paired",
"caption": "Mike Tyson's Punch Out",
"LR": "https://imgsli.com/i/0c4c242e-708b-4c47-b2a9-94f306a1e805.jpg",
"SR": "https://imgsli.com/i/bbe1d2b1-e24b-44d1-a765-da5c1dffbd87.jpg"
},
{
"type": "paired",
"caption": "Legacy of the Wizard (VHS)",
"LR": "https://imgsli.com/i/82831713-fa5d-43ac-8ec4-5f1aa28a715a.jpg",
"SR": "https://imgsli.com/i/a9351ae6-b095-46f7-b32f-a7f2d9dc941d.jpg"
},
{
"type": "paired",
"caption": "Defender of the Crown (RF)",
"LR": "https://imgsli.com/i/61c72f7a-efd5-48af-ad72-3114c654ecd9.jpg",
"SR": "https://imgsli.com/i/9f76ef1c-8660-43a6-b080-9d82ad9401ff.jpg"
},
{
"type": "paired",
"caption": "Ninja Gaiden (Game Center CX)",
"LR": "https://imgsli.com/i/571c1b9a-a2be-4c10-a080-3b62667a5b94.jpg",
"SR": "https://imgsli.com/i/a2d46549-bea4-4683-9ed7-3543afe99a06.jpg"
},
{
"type": "paired",
"caption": "Mega Man II (The Wizard)",
"LR": "https://imgsli.com/i/9c3f1b24-4ac4-4b93-8e3d-df04a926bad2.jpg",
"SR": "https://imgsli.com/i/cb1a0f83-3851-4a0f-85ed-3edcd9e1b426.jpg"
}
]
}
1 change: 1 addition & 0 deletions data/models/1x-NES-Composite-2-RGB.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
"tags": [
"debanding",
"denoise",
"game-screenshots",
"restoration",
"video-frame"
],
Expand Down
67 changes: 67 additions & 0 deletions data/models/1x-Sega-Genesis-Cleanup-Small.json
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,67 @@
{
"name": "Sega Genesis Cleanup-Small",
"author": "pokepress",
"license": "MIT",
"tags": [
"compression-removal",
"denoise",
"game-screenshots",
"restoration",
"video-frame"
],
"description": "Takes composite/RF/RGB/VHS Genesis footage and attempts to restore it to RGB quality. Assumes footage has been properly deinterlaced via field duplication from 240p to 480p/720p/etc. Note that:\n\n* All footage was captured in 240p/480p/720p NTSC.\n* Ground truth RGB footage was captured via a triple-bypassed Model 2 Genesis with new capacitors via the Retrotink 2x or GBS Control. Other footage was captured from various Model 1 Genesis consoles (with and without the \"High Definition Graphics\" mark).\n* The model was trained exclusively on individual frames, so it can't fix things like dropouts.\n* Should help fix color, blur, and jailbars.\n* Cannot un-dither fully blended pixels.\n\nCompared to the OmniSR variant of this model, this version runs more quickly (despite the larger model file size), and should do a good job on footage captured directly from the console using composite or RGB. If your footage is from a VHS tape or was captured over RF, the slower OmniSR model is probably a better choice.\n\nRevision History:\n\n1.0.0 (09/16/2025): Initial release.",
"date": "2025-09-16",
"architecture": "compact",
"size": null,
"scale": 1,
"inputChannels": 3,
"outputChannels": 3,
"resources": [
{
"platform": "pytorch",
"type": "pth",
"size": 8025883,
"sha256": "931df9a7620af7f8d735fd7e05f66a8743132506a3210a87c214f89ea72194dc",
"urls": [
"https://www.mediafire.com/file/f99yw9jczm0w1ju/Genesis_Cleanup_compact.pth/file"
]
}
],
"trainingIterations": 147608,
"trainingBatchSize": 8,
"trainingOTF": false,
"dataset": "Screenshots from various Genesis games.",
"datasetSize": 115,
"images": [
{
"type": "paired",
"caption": "Sonic 2 (Composite)",
"LR": "https://imgsli.com/i/cb5210fa-b962-4b66-b891-586a4dd73b3d.jpg",
"SR": "https://imgsli.com/i/cb9d6c04-012a-4e9a-86ba-7e9666cf6f66.jpg"
},
{
"type": "paired",
"caption": "T2: The Arcade Game (VHS)",
"LR": "https://imgsli.com/i/b7dd7f8a-161c-44e6-87e0-60f80c11c2cf.jpg",
"SR": "https://imgsli.com/i/c3d6ec1a-7a04-4033-95a0-a11959a5edd2.jpg"
},
{
"type": "paired",
"caption": "Tower Bridge (RF)",
"LR": "https://imgsli.com/i/9564a712-79b2-48c1-8043-18885d9e0640.jpg",
"SR": "https://imgsli.com/i/2ee8ddd3-ddf5-4d1b-a167-cb0d73091df4.jpg"
},
{
"type": "paired",
"caption": "Super Fantasy Zone (GameCenter CX)",
"LR": "https://imgsli.com/i/42aad9f2-0d65-485c-9e8d-7b07bb7a2583.jpg",
"SR": "https://imgsli.com/i/80922473-0dd3-405f-a5d7-a0db31d8940c.jpg"
},
{
"type": "paired",
"caption": "Sonic 2 (Model 1 RGB)",
"LR": "https://imgsli.com/i/70a80c2b-41d3-48c5-a040-704f72833fae.jpg",
"SR": "https://imgsli.com/i/bcbb28ef-91e7-4d74-8b05-0acfef62338d.jpg"
}
]
}
30 changes: 18 additions & 12 deletions data/models/2x-DigitalPokemon-l.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@
"cartoon",
"general-upscaler"
],
"description": "This model is designed to upscale the standard definition digital era of the Pokémon anime, which runs from late season 5 (Master Quest) to early season 12 (Galactic Battles). During this time, the show was animated digitally in a 4:3 ratio. This process was also used for Mewtwo Returns, most of Pokémon Chronicles, and the Mystery Dungeon specials.\n\nAdvice/Known Limitations:\n* This OmniSR model can occasionally produce black frames when run in fp16 mode. This seems to be more common in the TPCi era (seasons 9 and later). The issue is sporadic enough that it probably makes sense to do a first pass in fp16, then re-upscale any affected shots in fp32.\n* I recommend using QTGMC on a preset of \"Slow\" or slower for deinterlacing. While the show is primarily animated at 12/24 fps, some elements like backgrounds are animated at a full 60i.\n* The model is not great at handling fonts, particularly the italicized text in the episode credits. This is despite including font images in the training data,.",
"date": "2025-01-05",
"description": "This model is designed to upscale the standard definition digital era of the Pokémon anime, which runs from late season 5 (Master Quest) to early season 12 (Galactic Battles). During this time, the show was animated digitally in a 4:3 ratio. This process was also used for Mewtwo Returns, most of Pokémon Chronicles, and the Mystery Dungeon specials.\n\nAdvice/Known Limitations:\n* This OmniSR model can occasionally produce black frames when run in fp16 mode. This seems to be more common in the TPCi era (seasons 9 and later). The issue is sporadic enough that it probably makes sense to do a first pass in fp16, then re-upscale any affected shots in fp32.\n* I recommend using QTGMC on a preset of \"Slow\" or slower for deinterlacing. While the show is primarily animated at 12/24 fps, some elements like backgrounds are animated at a full 60i, particularly for the Diamond & Pearl era, which increased the base animation frame rate to 15 fps.\n* The model is not great at handling fonts, particularly the italicized text in the episode credits. This is despite including font images in the training data.\n\nVersion History:\n\n* 1.1.0 (9/16/2025): Added images from SD portion of season 12, augmented training to better handle slightly resized images.\n* 1.0.0: Initial Release",
"date": "2025-09-16",
"architecture": "omnisr",
"size": null,
"scale": 2,
Expand All @@ -18,42 +18,48 @@
{
"platform": "pytorch",
"type": "pth",
"size": 3244282,
"sha256": "ff5e8f803b6153b0c16ba246c5579524a978d16a90545d2736603703d9d67a2a",
"size": 3244367,
"sha256": "995824525ec8f858911d44be3e25ab4727acb68ceae4e68587234b0dfd553c4a",
"urls": [
"https://www.mediafire.com/file/gckgn1299krt2so/digital_pokemon_omnisr.pth/file"
"https://www.mediafire.com/file/yfedbuy8h1jpjlf/digital_pokemon_omnisr_1_1_0.pth/file"
]
}
],
"trainingIterations": 465000,
"trainingIterations": 510000,
"trainingHRSize": 64,
"trainingOTF": false,
"dataset": "Screenshots from seasons 6-11 of the English Pokémon dub, plus some manually created images for fonts.",
"datasetSize": 74,
"dataset": "Screenshots from seasons 6-12 of the English Pokémon dub, plus some manually created images for fonts.",
"datasetSize": 80,
"images": [
{
"type": "paired",
"caption": "A Bite to Remember",
"LR": "https://imgsli.com/i/9ae50058-ac4a-415a-a243-8f924015fd46.jpg",
"SR": "https://imgsli.com/i/d96a8dcd-5b70-464f-8b45-2ed443f8fa3d.jpg"
"SR": "https://imgsli.com/i/7208b898-7087-4b3f-b7e5-e0de26b7bee3.jpg"
},
{
"type": "paired",
"caption": "Going, Going, Yawn Title Card",
"LR": "https://imgsli.com/i/64ed6a26-8fb9-4849-996d-2d240b3eda97.jpg",
"SR": "https://imgsli.com/i/aea9dd40-c261-41d4-93d0-bea5f702700b.jpg"
"SR": "https://imgsli.com/i/149ea0dc-4a0d-4054-b303-029a637b84de.jpg"
},
{
"type": "paired",
"caption": "Fear Factor Phony!",
"LR": "https://imgsli.com/i/165ae236-385e-4519-b062-d7e8ec21d797.jpg",
"SR": "https://imgsli.com/i/fcbae9aa-dd0d-4c50-9071-86c8299f5b25.jpg"
"SR": "https://imgsli.com/i/ac297d41-2be8-4589-9faa-7bfcddcdc73e.jpg"
},
{
"type": "paired",
"caption": "Once There Were Greenfields",
"LR": "https://imgsli.com/i/91270773-c8e9-4ea3-a679-ce1acde79b60.jpg",
"SR": "https://imgsli.com/i/1498de96-c924-4fba-b7cb-3b128e765d05.jpg"
"SR": "https://imgsli.com/i/f336ac86-33a2-4ba1-b4d4-d676e637ee12.jpg"
},
{
"type": "paired",
"caption": "Cheers on Castaways Isle!",
"LR": "https://imgsli.com/i/f71c330d-8a10-4285-ada6-d9bcad0429c7.jpg",
"SR": "https://imgsli.com/i/768b6012-dc7c-42cd-a369-b25c1ba736e0.jpg"
}
]
}
28 changes: 17 additions & 11 deletions data/models/2x-DigitalPokemon-s.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
"cartoon",
"general-upscaler"
],
"description": "This model is designed to upscale the standard definition digital era of the Pokémon anime, which runs from late season 5 (Master Quest) to early season 12 (Galactic Battles). During this time, the show was animated digitally in a 4:3 ratio. This process was also used for Mewtwo Returns, most of Pokémon Chronicles, and the Mystery Dungeon specials.\n\nAdvice/Known Limitations:\n* I recommend using QTGMC on a preset of \"Slow\" or slower for deinterlacing. While the show is primarily animated at 12/24 fps, some elements like backgrounds are animated at a full 60i.\n* The model is not great at handling fonts, particularly the italicized text in the episode credits. This is despite including font images in the training data,.",
"description": "This model is designed to upscale the standard definition digital era of the Pokémon anime, which runs from late season 5 (Master Quest) to early season 12 (Galactic Battles). During this time, the show was animated digitally in a 4:3 ratio. This process was also used for Mewtwo Returns, most of Pokémon Chronicles, and the Mystery Dungeon specials.\n\nAdvice/Known Limitations:\n* I recommend using QTGMC on a preset of \"Slow\" or slower for deinterlacing. While the show is primarily animated at 12/24 fps, some elements like backgrounds are animated at a full 60i, particularly for the Diamond & Pearl era, which increased the base animation frame rate to 15 fps.\n* The model is not great at handling fonts, particularly the italicized text in the episode credits. This is despite including font images in the training data.\n\nVersion History:\n\n* 1.1.0 (9/16/2025): Added images from SD portion of season 12, augmented training to better handle slightly resized images.\n* 1.0.0: Initial Release",
"date": "2025-01-05",
"architecture": "compact",
"size": null,
Expand All @@ -18,42 +18,48 @@
{
"platform": "pytorch",
"type": "pth",
"size": 2418324,
"sha256": "bfa3409a2a2306314645c65f6660cff2077bde1ba9dbbbf08248ecbace7ade3a",
"size": 2418672,
"sha256": "c2df81fcfa5f7a892be31859d4adec8a73da60b0a34590d2e2a82a0f971df451",
"urls": [
"https://www.mediafire.com/file/37n5iwfkrpqd3ef/digital_pokemon_compact.pth/file"
"https://www.mediafire.com/file/5kpf1yg7bbw21ad/digital_pokemon_compact_1_1_0.pth/file"
]
}
],
"trainingIterations": 230000,
"trainingIterations": 280000,
"trainingHRSize": 64,
"trainingOTF": false,
"dataset": "Screenshots from seasons 6-11 of the English Pokémon dub, plus some manually created images for fonts.",
"datasetSize": 74,
"dataset": "Screenshots from seasons 6-12 of the English Pokémon dub, plus some manually created images for fonts.",
"datasetSize": 80,
"images": [
{
"type": "paired",
"caption": "A Bite to Remember",
"LR": "https://imgsli.com/i/db8243fa-875a-410e-a3cd-fafbb3af9cb1.jpg",
"SR": "https://imgsli.com/i/08e7cfd5-a764-4db4-9184-a9392504af3c.jpg"
"SR": "https://imgsli.com/i/9b072610-df0d-4079-a647-a515cc25ec8f.jpg"
},
{
"type": "paired",
"caption": "Going, Going, Yawn Title Card",
"LR": "https://imgsli.com/i/d36cf8f0-7755-443e-b248-c06e2a7c9449.jpg",
"SR": "https://imgsli.com/i/bdd8af00-b6c0-4a53-8465-bfa82c5fe0c3.jpg"
"SR": "https://imgsli.com/i/d276a501-5e96-42c6-98f0-72664e97193e.jpg"
},
{
"type": "paired",
"caption": "Fear Factor Phony!",
"LR": "https://imgsli.com/i/2d6cacbe-91e9-4564-a197-9a86d9ed4c60.jpg",
"SR": "https://imgsli.com/i/e7623bdb-6aec-4b83-bb90-8e7b0a4a63fd.jpg"
"SR": "https://imgsli.com/i/cf9d83e5-6d27-417b-bdfd-4c577e286bb8.jpg"
},
{
"type": "paired",
"caption": "Once There Were Greenfields",
"LR": "https://imgsli.com/i/5a326f65-f23f-4eee-9538-94eba40d0c0b.jpg",
"SR": "https://imgsli.com/i/93ca0d6e-28ac-4eef-9227-8e828b90a045.jpg"
"SR": "https://imgsli.com/i/be3a0b9a-2b50-4ffd-b951-a90517bd67c1.jpg"
},
{
"type": "paired",
"caption": "Cheers on Castaways Isle!",
"LR": "https://imgsli.com/i/ef6de563-3a65-450e-ade3-387bcba2212c.jpg",
"SR": "https://imgsli.com/i/368cd5de-fa07-4ae8-97fb-f73e31321b59.jpg"
}
]
}