Fixed pseudo F-measure metric in README.md
Browse files
README.md
CHANGED
|
@@ -20,11 +20,13 @@ This model is a fine-tuned version of [nvidia/segformer-b3-finetuned-cityscapes-
|
|
| 20 |
It achieves the following results on the evaluation set on DIBCO metrics:
|
| 21 |
- loss: 0.1017
|
| 22 |
- F-measure: 0.9776
|
| 23 |
-
-
|
| 24 |
- PSNR: 14.5040
|
| 25 |
- DRD: 5.3749
|
| 26 |
|
| 27 |
-
|
|
|
|
|
|
|
| 28 |
|
| 29 |
**Warning:** This model only accepts images with a resolution of 640 due to compute constraints on Colab free tier during training.
|
| 30 |
|
|
@@ -58,7 +60,7 @@ The following hyperparameters were used during training:
|
|
| 58 |
|
| 59 |
### Training results
|
| 60 |
|
| 61 |
-
| training loss | epoch | step | validation loss | F-measure |
|
| 62 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:-------:|:--------:|
|
| 63 |
| 0.6667 | 1.03 | 10 | 0.6683 | 0.7127 | 0.6831 | 4.8248 | 107.2894 |
|
| 64 |
| 0.6371 | 2.05 | 20 | 0.6390 | 0.8173 | 0.7360 | 6.1079 | 69.7770 |
|
|
|
|
| 20 |
It achieves the following results on the evaluation set on DIBCO metrics:
|
| 21 |
- loss: 0.1017
|
| 22 |
- F-measure: 0.9776
|
| 23 |
+
- pseudo F-measure: 0.9531
|
| 24 |
- PSNR: 14.5040
|
| 25 |
- DRD: 5.3749
|
| 26 |
|
| 27 |
+
where PSNR stands for peak signal-to-noise ratio and DND for distance reciprocal distortion.
|
| 28 |
+
|
| 29 |
+
For more information on DIBCO metrics, see the 2017 introductory [paper](https://ieeexplore.ieee.org/document/8270159).
|
| 30 |
|
| 31 |
**Warning:** This model only accepts images with a resolution of 640 due to compute constraints on Colab free tier during training.
|
| 32 |
|
|
|
|
| 60 |
|
| 61 |
### Training results
|
| 62 |
|
| 63 |
+
| training loss | epoch | step | validation loss | F-measure | pseudo F-measure | PSNR | DRD |
|
| 64 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:-------:|:--------:|
|
| 65 |
| 0.6667 | 1.03 | 10 | 0.6683 | 0.7127 | 0.6831 | 4.8248 | 107.2894 |
|
| 66 |
| 0.6371 | 2.05 | 20 | 0.6390 | 0.8173 | 0.7360 | 6.1079 | 69.7770 |
|