Onegafer commited on
Commit
4dae927
·
1 Parent(s): bf9d8bc

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -11
README.md CHANGED
@@ -16,14 +16,14 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [nvidia/mit-b2](https://huggingface.co/nvidia/mit-b2) on the Onegafer/vehicle_segmentation dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.2948
20
- - Mean Iou: 0.2111
21
- - Mean Accuracy: 0.4222
22
- - Overall Accuracy: 0.4222
23
- - Accuracy No: nan
24
- - Accuracy Yes: 0.4222
25
- - Iou No: 0.0
26
- - Iou Yes: 0.4222
27
 
28
  ## Model description
29
 
@@ -52,9 +52,9 @@ The following hyperparameters were used during training:
52
 
53
  ### Training results
54
 
55
- | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy No | Accuracy Yes | Iou No | Iou Yes |
56
- |:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:-----------:|:------------:|:------:|:-------:|
57
- | 0.3775 | 0.16 | 20 | 0.2948 | 0.2111 | 0.4222 | 0.4222 | nan | 0.4222 | 0.0 | 0.4222 |
58
 
59
 
60
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [nvidia/mit-b2](https://huggingface.co/nvidia/mit-b2) on the Onegafer/vehicle_segmentation dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.3350
20
+ - Mean Iou: 0.3373
21
+ - Mean Accuracy: 0.6746
22
+ - Overall Accuracy: 0.6746
23
+ - Accuracy Background: nan
24
+ - Accuracy Windows: 0.6746
25
+ - Iou Background: 0.0
26
+ - Iou Windows: 0.6746
27
 
28
  ## Model description
29
 
 
52
 
53
  ### Training results
54
 
55
+ | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Background | Accuracy Windows | Iou Background | Iou Windows |
56
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:-------------------:|:----------------:|:--------------:|:-----------:|
57
+ | 0.4394 | 0.16 | 20 | 0.3350 | 0.3373 | 0.6746 | 0.6746 | nan | 0.6746 | 0.0 | 0.6746 |
58
 
59
 
60
  ### Framework versions