|
--- |
|
license: cc-by-4.0 |
|
pipeline_tag: image-to-image |
|
tags: |
|
- pytorch |
|
- super-resolution |
|
- pretrain |
|
--- |
|
|
|
[Link to Github Release](https://github.com/Phhofm/models/releases/tag/4xDRCT-mssim-pretrains) |
|
|
|
Since no [DRCT](https://github.com/ming053l/DRCT/tree/main) model releases exist yet, I am releasing my drct-s, drct and drct-l pretrains. |
|
These can be used to speed up and stabilize early training stages when training new drct models. |
|
Trained with mssim (and augs, and color and lumaloss) on downscaled [nomosuni](https://github.com/muslll/neosr/tree/0ecbeb98dcefd97801da0ea39ca2cd03a87fce21?tab=readme-ov-file#datasets). |
|
Training files are also provided additionally. |
|
|
|
|
|
## 4xmssim_drct-s_pretrain |
|
|
|
**Scale:** 4 |
|
**Architecture:** [DRCT](https://github.com/ming053l/DRCT) |
|
**Architecture Option:** [DRCT-S](https://github.com/muslll/neosr/blob/0ecbeb98dcefd97801da0ea39ca2cd03a87fce21/neosr/archs/drct_arch.py#L923) |
|
|
|
**Author:** Philip Hofmann |
|
**License:** CC-BY-0.4 |
|
**Purpose:** Pretrained |
|
**Subject:** Anime, Photography |
|
**Input Type:** Images |
|
**Release Date:** 28.04.2024 |
|
|
|
**Dataset:** [nomos_uni](https://github.com/muslll/neosr?tab=readme-ov-file#datasets) |
|
**Dataset Size:** 2989 |
|
**OTF (on the fly augmentations):** No |
|
**Pretrained Model:** None (=From Scratch) |
|
**Iterations:** 75'000 |
|
**Batch Size:** 6 |
|
**GT Size:** 128-320 |
|
|
|
**Description:** A pretrain to start drct-s model training, since there are no official released drct pretrains yet I trained these myself and release them here. |
|
|
|
--- |
|
|
|
## 4xmssim_drct_pretrain |
|
|
|
**Scale:** 4 |
|
**Architecture:** [DRCT](https://github.com/ming053l/DRCT) |
|
**Architecture Option:** [DRCT](https://github.com/ming053l/DRCT/blob/8c13e63135f494913cd504c073e41ef52250d1d4/options/train/train_DRCT_SRx4_ImageNet_from_scratch.yml#L55) |
|
|
|
**Author:** Philip Hofmann |
|
**License:** CC-BY-0.4 |
|
**Purpose:** Pretrained |
|
**Subject:** Anime, Photography |
|
**Input Type:** Images |
|
**Release Date:** 28.04.2024 |
|
|
|
**Dataset:** [nomos_uni](https://github.com/muslll/neosr?tab=readme-ov-file#datasets) |
|
**Dataset Size:** 2989 |
|
**OTF (on the fly augmentations):** No |
|
**Pretrained Model:** None (=From Scratch) |
|
**Iterations:** 95'000 |
|
**Batch Size:** 2-6 |
|
**GT Size:** 128-320 |
|
|
|
**Description:** A pretrain to start drct model training, since there are no official released drct pretrains yet I trained these myself and release them here. |
|
|
|
--- |
|
|
|
## 4xmssim_drct-l_pretrain |
|
|
|
**Scale:** 4 |
|
**Architecture:** [DRCT](https://github.com/ming053l/DRCT) |
|
**Architecture Option:** [DRCT-L](https://github.com/ming053l/DRCT/blob/8c13e63135f494913cd504c073e41ef52250d1d4/options/train/train_DRCT-L_SRx4_ImageNet_from_scratch.yml#L55) |
|
|
|
**Author:** Philip Hofmann |
|
**License:** CC-BY-0.4 |
|
**Purpose:** Pretrained |
|
**Subject:** Anime, Photography |
|
**Input Type:** Images |
|
**Release Date:** 28.04.2024 |
|
|
|
**Dataset:** [nomos_uni](https://github.com/muslll/neosr?tab=readme-ov-file#datasets) |
|
**Dataset Size:** 2989 |
|
**OTF (on the fly augmentations):** No |
|
**Pretrained Model:** None (=From Scratch) |
|
**Iterations:** 108'000 |
|
**Batch Size:** 2-6 |
|
**GT Size:** 128-320 |
|
|
|
**Description:** A pretrain to start drct-l model training, since there are no official released drct pretrains yet I trained these myself and release them here. |
|
|
|
--- |
|
|
|
**Showcase:** |
|
|
|
[slow.pics](https://slow.pics/c/9lmWhRto) |
|
|
|
![Example1](https://github.com/Phhofm/models/assets/14755670/5fa94ef9-87dc-41d0-937f-9024c6121490) |
|
![Example2](https://github.com/Phhofm/models/assets/14755670/7dc829bf-7a93-4571-81bf-74cae125b907) |
|
![Example3](https://github.com/Phhofm/models/assets/14755670/688fbe5c-f78a-4e8a-b833-32075d932a7c) |
|
![Example4](https://github.com/Phhofm/models/assets/14755670/a1c201af-4278-4870-b6ca-16017a8ad24e) |
|
![Example5](https://github.com/Phhofm/models/assets/14755670/4b07d6ba-7503-4da9-9915-d55dbe71581d) |
|
![Example6](https://github.com/Phhofm/models/assets/14755670/7fe8813f-a0a4-482e-8c48-a3f330039551) |
|
|
|
|
|
|