Skip to content

Commit a515f7a

Browse files
authored
Update digital pokemon models, add compact verions of NES/Genesis models. (#490)
1 parent 64d79a1 commit a515f7a

5 files changed

Lines changed: 171 additions & 23 deletions
Lines changed: 68 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,68 @@
1+
{
2+
"name": "NES-Composite-2-RGB-Small",
3+
"author": "pokepress",
4+
"license": "MIT",
5+
"tags": [
6+
"debanding",
7+
"denoise",
8+
"game-screenshots",
9+
"restoration",
10+
"video-frame"
11+
],
12+
"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",
13+
"date": "2025-09-17",
14+
"architecture": "compact",
15+
"size": null,
16+
"scale": 1,
17+
"inputChannels": 3,
18+
"outputChannels": 3,
19+
"resources": [
20+
{
21+
"platform": "pytorch",
22+
"type": "pth",
23+
"size": 8025476,
24+
"sha256": "6ce8754f660dd44c7c98e5c15e5125eaeb7a593d003c29b67fbf351099fcae73",
25+
"urls": [
26+
"https://www.mediafire.com/file/xuzy50qsayrkpxq/NES_Composite_To_RGB_compact_1_5_0.pth/file"
27+
]
28+
}
29+
],
30+
"trainingIterations": 236330,
31+
"trainingBatchSize": 8,
32+
"trainingHRSize": 128,
33+
"trainingOTF": false,
34+
"dataset": "Screenshots of NES games",
35+
"datasetSize": 198,
36+
"images": [
37+
{
38+
"type": "paired",
39+
"caption": "Mike Tyson's Punch Out",
40+
"LR": "https://imgsli.com/i/0c4c242e-708b-4c47-b2a9-94f306a1e805.jpg",
41+
"SR": "https://imgsli.com/i/bbe1d2b1-e24b-44d1-a765-da5c1dffbd87.jpg"
42+
},
43+
{
44+
"type": "paired",
45+
"caption": "Legacy of the Wizard (VHS)",
46+
"LR": "https://imgsli.com/i/82831713-fa5d-43ac-8ec4-5f1aa28a715a.jpg",
47+
"SR": "https://imgsli.com/i/a9351ae6-b095-46f7-b32f-a7f2d9dc941d.jpg"
48+
},
49+
{
50+
"type": "paired",
51+
"caption": "Defender of the Crown (RF)",
52+
"LR": "https://imgsli.com/i/61c72f7a-efd5-48af-ad72-3114c654ecd9.jpg",
53+
"SR": "https://imgsli.com/i/9f76ef1c-8660-43a6-b080-9d82ad9401ff.jpg"
54+
},
55+
{
56+
"type": "paired",
57+
"caption": "Ninja Gaiden (Game Center CX)",
58+
"LR": "https://imgsli.com/i/571c1b9a-a2be-4c10-a080-3b62667a5b94.jpg",
59+
"SR": "https://imgsli.com/i/a2d46549-bea4-4683-9ed7-3543afe99a06.jpg"
60+
},
61+
{
62+
"type": "paired",
63+
"caption": "Mega Man II (The Wizard)",
64+
"LR": "https://imgsli.com/i/9c3f1b24-4ac4-4b93-8e3d-df04a926bad2.jpg",
65+
"SR": "https://imgsli.com/i/cb1a0f83-3851-4a0f-85ed-3edcd9e1b426.jpg"
66+
}
67+
]
68+
}

data/models/1x-NES-Composite-2-RGB.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
"tags": [
66
"debanding",
77
"denoise",
8+
"game-screenshots",
89
"restoration",
910
"video-frame"
1011
],
Lines changed: 67 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,67 @@
1+
{
2+
"name": "Sega Genesis Cleanup-Small",
3+
"author": "pokepress",
4+
"license": "MIT",
5+
"tags": [
6+
"compression-removal",
7+
"denoise",
8+
"game-screenshots",
9+
"restoration",
10+
"video-frame"
11+
],
12+
"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.",
13+
"date": "2025-09-16",
14+
"architecture": "compact",
15+
"size": null,
16+
"scale": 1,
17+
"inputChannels": 3,
18+
"outputChannels": 3,
19+
"resources": [
20+
{
21+
"platform": "pytorch",
22+
"type": "pth",
23+
"size": 8025883,
24+
"sha256": "931df9a7620af7f8d735fd7e05f66a8743132506a3210a87c214f89ea72194dc",
25+
"urls": [
26+
"https://www.mediafire.com/file/f99yw9jczm0w1ju/Genesis_Cleanup_compact.pth/file"
27+
]
28+
}
29+
],
30+
"trainingIterations": 147608,
31+
"trainingBatchSize": 8,
32+
"trainingOTF": false,
33+
"dataset": "Screenshots from various Genesis games.",
34+
"datasetSize": 115,
35+
"images": [
36+
{
37+
"type": "paired",
38+
"caption": "Sonic 2 (Composite)",
39+
"LR": "https://imgsli.com/i/cb5210fa-b962-4b66-b891-586a4dd73b3d.jpg",
40+
"SR": "https://imgsli.com/i/cb9d6c04-012a-4e9a-86ba-7e9666cf6f66.jpg"
41+
},
42+
{
43+
"type": "paired",
44+
"caption": "T2: The Arcade Game (VHS)",
45+
"LR": "https://imgsli.com/i/b7dd7f8a-161c-44e6-87e0-60f80c11c2cf.jpg",
46+
"SR": "https://imgsli.com/i/c3d6ec1a-7a04-4033-95a0-a11959a5edd2.jpg"
47+
},
48+
{
49+
"type": "paired",
50+
"caption": "Tower Bridge (RF)",
51+
"LR": "https://imgsli.com/i/9564a712-79b2-48c1-8043-18885d9e0640.jpg",
52+
"SR": "https://imgsli.com/i/2ee8ddd3-ddf5-4d1b-a167-cb0d73091df4.jpg"
53+
},
54+
{
55+
"type": "paired",
56+
"caption": "Super Fantasy Zone (GameCenter CX)",
57+
"LR": "https://imgsli.com/i/42aad9f2-0d65-485c-9e8d-7b07bb7a2583.jpg",
58+
"SR": "https://imgsli.com/i/80922473-0dd3-405f-a5d7-a0db31d8940c.jpg"
59+
},
60+
{
61+
"type": "paired",
62+
"caption": "Sonic 2 (Model 1 RGB)",
63+
"LR": "https://imgsli.com/i/70a80c2b-41d3-48c5-a040-704f72833fae.jpg",
64+
"SR": "https://imgsli.com/i/bcbb28ef-91e7-4d74-8b05-0acfef62338d.jpg"
65+
}
66+
]
67+
}

data/models/2x-DigitalPokemon-l.json

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@
77
"cartoon",
88
"general-upscaler"
99
],
10-
"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,.",
11-
"date": "2025-01-05",
10+
"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",
11+
"date": "2025-09-16",
1212
"architecture": "omnisr",
1313
"size": null,
1414
"scale": 2,
@@ -18,42 +18,48 @@
1818
{
1919
"platform": "pytorch",
2020
"type": "pth",
21-
"size": 3244282,
22-
"sha256": "ff5e8f803b6153b0c16ba246c5579524a978d16a90545d2736603703d9d67a2a",
21+
"size": 3244367,
22+
"sha256": "995824525ec8f858911d44be3e25ab4727acb68ceae4e68587234b0dfd553c4a",
2323
"urls": [
24-
"https://www.mediafire.com/file/gckgn1299krt2so/digital_pokemon_omnisr.pth/file"
24+
"https://www.mediafire.com/file/yfedbuy8h1jpjlf/digital_pokemon_omnisr_1_1_0.pth/file"
2525
]
2626
}
2727
],
28-
"trainingIterations": 465000,
28+
"trainingIterations": 510000,
2929
"trainingHRSize": 64,
3030
"trainingOTF": false,
31-
"dataset": "Screenshots from seasons 6-11 of the English Pokémon dub, plus some manually created images for fonts.",
32-
"datasetSize": 74,
31+
"dataset": "Screenshots from seasons 6-12 of the English Pokémon dub, plus some manually created images for fonts.",
32+
"datasetSize": 80,
3333
"images": [
3434
{
3535
"type": "paired",
3636
"caption": "A Bite to Remember",
3737
"LR": "https://imgsli.com/i/9ae50058-ac4a-415a-a243-8f924015fd46.jpg",
38-
"SR": "https://imgsli.com/i/d96a8dcd-5b70-464f-8b45-2ed443f8fa3d.jpg"
38+
"SR": "https://imgsli.com/i/7208b898-7087-4b3f-b7e5-e0de26b7bee3.jpg"
3939
},
4040
{
4141
"type": "paired",
4242
"caption": "Going, Going, Yawn Title Card",
4343
"LR": "https://imgsli.com/i/64ed6a26-8fb9-4849-996d-2d240b3eda97.jpg",
44-
"SR": "https://imgsli.com/i/aea9dd40-c261-41d4-93d0-bea5f702700b.jpg"
44+
"SR": "https://imgsli.com/i/149ea0dc-4a0d-4054-b303-029a637b84de.jpg"
4545
},
4646
{
4747
"type": "paired",
4848
"caption": "Fear Factor Phony!",
4949
"LR": "https://imgsli.com/i/165ae236-385e-4519-b062-d7e8ec21d797.jpg",
50-
"SR": "https://imgsli.com/i/fcbae9aa-dd0d-4c50-9071-86c8299f5b25.jpg"
50+
"SR": "https://imgsli.com/i/ac297d41-2be8-4589-9faa-7bfcddcdc73e.jpg"
5151
},
5252
{
5353
"type": "paired",
5454
"caption": "Once There Were Greenfields",
5555
"LR": "https://imgsli.com/i/91270773-c8e9-4ea3-a679-ce1acde79b60.jpg",
56-
"SR": "https://imgsli.com/i/1498de96-c924-4fba-b7cb-3b128e765d05.jpg"
56+
"SR": "https://imgsli.com/i/f336ac86-33a2-4ba1-b4d4-d676e637ee12.jpg"
57+
},
58+
{
59+
"type": "paired",
60+
"caption": "Cheers on Castaways Isle!",
61+
"LR": "https://imgsli.com/i/f71c330d-8a10-4285-ada6-d9bcad0429c7.jpg",
62+
"SR": "https://imgsli.com/i/768b6012-dc7c-42cd-a369-b25c1ba736e0.jpg"
5763
}
5864
]
5965
}

data/models/2x-DigitalPokemon-s.json

Lines changed: 17 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
"cartoon",
88
"general-upscaler"
99
],
10-
"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,.",
10+
"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",
1111
"date": "2025-01-05",
1212
"architecture": "compact",
1313
"size": null,
@@ -18,42 +18,48 @@
1818
{
1919
"platform": "pytorch",
2020
"type": "pth",
21-
"size": 2418324,
22-
"sha256": "bfa3409a2a2306314645c65f6660cff2077bde1ba9dbbbf08248ecbace7ade3a",
21+
"size": 2418672,
22+
"sha256": "c2df81fcfa5f7a892be31859d4adec8a73da60b0a34590d2e2a82a0f971df451",
2323
"urls": [
24-
"https://www.mediafire.com/file/37n5iwfkrpqd3ef/digital_pokemon_compact.pth/file"
24+
"https://www.mediafire.com/file/5kpf1yg7bbw21ad/digital_pokemon_compact_1_1_0.pth/file"
2525
]
2626
}
2727
],
28-
"trainingIterations": 230000,
28+
"trainingIterations": 280000,
2929
"trainingHRSize": 64,
3030
"trainingOTF": false,
31-
"dataset": "Screenshots from seasons 6-11 of the English Pokémon dub, plus some manually created images for fonts.",
32-
"datasetSize": 74,
31+
"dataset": "Screenshots from seasons 6-12 of the English Pokémon dub, plus some manually created images for fonts.",
32+
"datasetSize": 80,
3333
"images": [
3434
{
3535
"type": "paired",
3636
"caption": "A Bite to Remember",
3737
"LR": "https://imgsli.com/i/db8243fa-875a-410e-a3cd-fafbb3af9cb1.jpg",
38-
"SR": "https://imgsli.com/i/08e7cfd5-a764-4db4-9184-a9392504af3c.jpg"
38+
"SR": "https://imgsli.com/i/9b072610-df0d-4079-a647-a515cc25ec8f.jpg"
3939
},
4040
{
4141
"type": "paired",
4242
"caption": "Going, Going, Yawn Title Card",
4343
"LR": "https://imgsli.com/i/d36cf8f0-7755-443e-b248-c06e2a7c9449.jpg",
44-
"SR": "https://imgsli.com/i/bdd8af00-b6c0-4a53-8465-bfa82c5fe0c3.jpg"
44+
"SR": "https://imgsli.com/i/d276a501-5e96-42c6-98f0-72664e97193e.jpg"
4545
},
4646
{
4747
"type": "paired",
4848
"caption": "Fear Factor Phony!",
4949
"LR": "https://imgsli.com/i/2d6cacbe-91e9-4564-a197-9a86d9ed4c60.jpg",
50-
"SR": "https://imgsli.com/i/e7623bdb-6aec-4b83-bb90-8e7b0a4a63fd.jpg"
50+
"SR": "https://imgsli.com/i/cf9d83e5-6d27-417b-bdfd-4c577e286bb8.jpg"
5151
},
5252
{
5353
"type": "paired",
5454
"caption": "Once There Were Greenfields",
5555
"LR": "https://imgsli.com/i/5a326f65-f23f-4eee-9538-94eba40d0c0b.jpg",
56-
"SR": "https://imgsli.com/i/93ca0d6e-28ac-4eef-9227-8e828b90a045.jpg"
56+
"SR": "https://imgsli.com/i/be3a0b9a-2b50-4ffd-b951-a90517bd67c1.jpg"
57+
},
58+
{
59+
"type": "paired",
60+
"caption": "Cheers on Castaways Isle!",
61+
"LR": "https://imgsli.com/i/ef6de563-3a65-450e-ade3-387bcba2212c.jpg",
62+
"SR": "https://imgsli.com/i/368cd5de-fa07-4ae8-97fb-f73e31321b59.jpg"
5763
}
5864
]
5965
}

0 commit comments

Comments
 (0)