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 |
|