|
--- |
|
license: cc-by-4.0 |
|
pipeline_tag: image-to-image |
|
tags: |
|
- pytorch |
|
- super-resolution |
|
- pretrain |
|
--- |
|
|
|
[Github Link](https://github.com/Phhofm/models/releases/tag/4xmssim_hma_pretrains) |
|
|
|
Since no official [HMA](https://github.com/korouuuuu/HMA) model releases exist yet, I am releasing my [hma](https://github.com/muslll/neosr/blob/a50b56de418c938425fd3606697f62d61dbddfc3/neosr/archs/hma_arch.py#L1148) and [hma_medium](https://github.com/muslll/neosr/blob/a50b56de418c938425fd3606697f62d61dbddfc3/neosr/archs/hma_arch.py#L1386) mssim pretrains. |
|
These can be used to speed up and stabilize early training stages when training new hma models. |
|
Trained with mssim on [nomosv2](https://github.com/muslll/neosr/tree/a50b56de418c938425fd3606697f62d61dbddfc3?tab=readme-ov-file#-datasets). |
|
|
|
|
|
## 4xmssim_hma_pretrain |
|
|
|
**Scale:** 4 |
|
**Architecture:** [HMA](https://github.com/korouuuuu/HMA) |
|
**Architecture Option:** [hma](https://github.com/muslll/neosr/blob/a50b56de418c938425fd3606697f62d61dbddfc3/neosr/archs/hma_arch.py#L1148) |
|
|
|
**Author:** Philip Hofmann |
|
**License:** CC-BY-0.4 |
|
**Purpose:** Pretrained |
|
**Subject:** Photography |
|
**Input Type:** Images |
|
**Release Date:** 19.07.2024 |
|
|
|
**Dataset:** [nomosv2](https://github.com/muslll/neosr/tree/a50b56de418c938425fd3606697f62d61dbddfc3?tab=readme-ov-file#-datasets) |
|
**Dataset Size:** 6000 |
|
**OTF (on the fly augmentations):** No |
|
**Pretrained Model:** None (=From Scratch) |
|
**Iterations:** 205'000 |
|
**Batch Size:** 4 |
|
**Patch Size:** 96 |
|
|
|
**Description:** A pretrain to start hma model training. |
|
|
|
--- |
|
|
|
## 4xmssim_hma_medium_pretrain |
|
|
|
**Scale:** 4 |
|
**Architecture:** [HMA](https://github.com/korouuuuu/HMA) |
|
**Architecture Option:** [hma_medium](https://github.com/muslll/neosr/blob/a50b56de418c938425fd3606697f62d61dbddfc3/neosr/archs/hma_arch.py#L1386) |
|
|
|
**Author:** Philip Hofmann |
|
**License:** CC-BY-0.4 |
|
**Purpose:** Pretrained |
|
**Subject:** Photography |
|
**Input Type:** Images |
|
**Release Date:** 19.07.2024 |
|
|
|
**Dataset:** [nomosv2](https://github.com/muslll/neosr/tree/a50b56de418c938425fd3606697f62d61dbddfc3?tab=readme-ov-file#-datasets) |
|
**Dataset Size:** 6000 |
|
**OTF (on the fly augmentations):** No |
|
**Pretrained Model:** None (=From Scratch) |
|
**Iterations:** 150'000 |
|
**Batch Size:** 4 |
|
**Patch Size:** 48 |
|
|
|
**Description:** A pretrain to start hma_medium model training. |
|
|
|
--- |
|
|
|
**Showcase:** |
|
|
|
[slow.pics](https://slow.pics/c/Eoq18VDx) |
|
|
|
 |
|
 |
|
 |
|
 |
|
|
|
|
