Skip to content
This repository was archived by the owner on Jan 7, 2025. It is now read-only.

Commit da81bcd

Browse files
authored
Merge pull request #205 from idealo/cfrancesco-patch-2
Update README.md
2 parents 64c9357 + 65cb373 commit da81bcd

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

README.md

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,36 @@ ISR is compatible with Python 3.6 and is distributed under the Apache 2.0 licens
3333
- [Maintainers](#maintainers)
3434
- [License](#copyright)
3535

36+
## Troubleshooting
37+
### Training not delivering good/patchy results
38+
When training your own model, start with only PSNR loss (50+ epochs, depending on the dataset) and only then introduce GANS and feature loss. This can be controlled by the loss weights argument.
39+
40+
This is just sample, you will need to tune these parameters.
41+
42+
PSNR only:
43+
```
44+
loss_weights = {
45+
'generator': 1.0,
46+
'feature_extractor': 0.0,
47+
'discriminator': 0.00
48+
}
49+
```
50+
51+
Later:
52+
```
53+
loss_weights = {
54+
'generator': 0.0,
55+
'feature_extractor': 0.0833,
56+
'discriminator': 0.01
57+
}
58+
```
59+
### Weights loading
60+
If you are having trouble loading your own weights or the pre-trained weights (`AttributeError: 'str' object has no attribute 'decode'`), try:
61+
```bash
62+
pip install 'h5py==2.10.0' --force-reinstall
63+
```
64+
[Issue](https://github.com/idealo/image-super-resolution/issues/197#issue-877826405)
65+
3666
## Pre-trained networks
3767

3868
The weights used to produced these images are available directly when creating the model object.

0 commit comments

Comments
 (0)