File size: 2,833 Bytes
be9d39a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
---
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)  

![Example1](https://github.com/user-attachments/assets/eb02bf8d-91bc-4442-81cb-6496f930dd0d)  
![Example2](https://github.com/user-attachments/assets/5ceec445-37fe-416e-af5c-c6d0e759ff72)  
![Example3](https://github.com/user-attachments/assets/39792fa6-250c-4fea-b4f8-4c9dbf80a4db)  
![Example4](https://github.com/user-attachments/assets/16f839af-ccfa-485a-bb2d-2635930a329f)