Update README file
Browse files
README.md
CHANGED
|
@@ -6,41 +6,41 @@ tags: []
|
|
| 6 |
## Original result
|
| 7 |
```
|
| 8 |
IoU metric: bbox
|
| 9 |
-
Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.
|
| 10 |
-
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.
|
| 11 |
-
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.
|
| 12 |
-
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.
|
| 13 |
-
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.
|
| 14 |
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = -1.000
|
| 15 |
-
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.
|
| 16 |
-
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.
|
| 17 |
-
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.
|
| 18 |
-
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.
|
| 19 |
-
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.
|
| 20 |
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = -1.000
|
| 21 |
```
|
| 22 |
|
| 23 |
## After training result
|
| 24 |
```
|
| 25 |
IoU metric: bbox
|
| 26 |
-
Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.
|
| 27 |
-
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.
|
| 28 |
-
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.
|
| 29 |
-
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.
|
| 30 |
-
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.
|
| 31 |
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = -1.000
|
| 32 |
-
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.
|
| 33 |
-
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.
|
| 34 |
-
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.
|
| 35 |
-
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.
|
| 36 |
-
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.
|
| 37 |
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = -1.000
|
| 38 |
```
|
| 39 |
|
| 40 |
## Config
|
| 41 |
- dataset: NIH
|
| 42 |
- original model: facebook/detr-resnet-50
|
| 43 |
-
- lr:
|
| 44 |
- dropout_rate: 0.1
|
| 45 |
- weight_decay: 0.05
|
| 46 |
- max_epochs: 20
|
|
@@ -49,27 +49,27 @@ IoU metric: bbox
|
|
| 49 |
## Logging
|
| 50 |
### Training process
|
| 51 |
```
|
| 52 |
-
{'validation_loss': tensor(2.
|
| 53 |
-
{'training_loss': tensor(
|
| 54 |
-
{'training_loss': tensor(
|
| 55 |
-
{'training_loss': tensor(0.
|
| 56 |
-
{'training_loss': tensor(0.
|
| 57 |
-
{'training_loss': tensor(0.
|
| 58 |
-
{'training_loss': tensor(
|
| 59 |
-
{'training_loss': tensor(
|
| 60 |
-
{'training_loss': tensor(
|
| 61 |
-
{'training_loss': tensor(
|
| 62 |
-
{'training_loss': tensor(
|
| 63 |
-
{'training_loss': tensor(
|
| 64 |
-
{'training_loss': tensor(
|
| 65 |
-
{'training_loss': tensor(
|
| 66 |
-
{'training_loss': tensor(2.
|
| 67 |
-
{'training_loss': tensor(
|
| 68 |
-
{'training_loss': tensor(
|
| 69 |
-
{'training_loss': tensor(
|
| 70 |
-
{'training_loss': tensor(
|
| 71 |
-
{'training_loss': tensor(
|
| 72 |
-
{'training_loss': tensor(
|
| 73 |
```
|
| 74 |
|
| 75 |
## Examples
|
|
|
|
| 6 |
## Original result
|
| 7 |
```
|
| 8 |
IoU metric: bbox
|
| 9 |
+
Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.044
|
| 10 |
+
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.056
|
| 11 |
+
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.051
|
| 12 |
+
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.157
|
| 13 |
+
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.030
|
| 14 |
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = -1.000
|
| 15 |
+
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.070
|
| 16 |
+
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.202
|
| 17 |
+
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.466
|
| 18 |
+
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.389
|
| 19 |
+
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.557
|
| 20 |
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = -1.000
|
| 21 |
```
|
| 22 |
|
| 23 |
## After training result
|
| 24 |
```
|
| 25 |
IoU metric: bbox
|
| 26 |
+
Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.001
|
| 27 |
+
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.010
|
| 28 |
+
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.000
|
| 29 |
+
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.031
|
| 30 |
+
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.009
|
| 31 |
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = -1.000
|
| 32 |
+
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.000
|
| 33 |
+
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.018
|
| 34 |
+
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.048
|
| 35 |
+
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.030
|
| 36 |
+
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.070
|
| 37 |
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = -1.000
|
| 38 |
```
|
| 39 |
|
| 40 |
## Config
|
| 41 |
- dataset: NIH
|
| 42 |
- original model: facebook/detr-resnet-50
|
| 43 |
+
- lr: 5e-05
|
| 44 |
- dropout_rate: 0.1
|
| 45 |
- weight_decay: 0.05
|
| 46 |
- max_epochs: 20
|
|
|
|
| 49 |
## Logging
|
| 50 |
### Training process
|
| 51 |
```
|
| 52 |
+
{'validation_loss': tensor(2.5684, device='cuda:0'), 'validation_loss_ce': tensor(0.6809, device='cuda:0'), 'validation_loss_bbox': tensor(0.1171, device='cuda:0'), 'validation_loss_giou': tensor(0.6509, device='cuda:0'), 'validation_cardinality_error': tensor(38.2500, device='cuda:0')}
|
| 53 |
+
{'training_loss': tensor(4.4525, device='cuda:0'), 'train_loss_ce': tensor(0.8844, device='cuda:0'), 'train_loss_bbox': tensor(0.3868, device='cuda:0'), 'train_loss_giou': tensor(0.8171, device='cuda:0'), 'train_cardinality_error': tensor(19., device='cuda:0'), 'validation_loss': tensor(1.2852, device='cuda:0'), 'validation_loss_ce': tensor(0.5324, device='cuda:0'), 'validation_loss_bbox': tensor(0.0360, device='cuda:0'), 'validation_loss_giou': tensor(0.2863, device='cuda:0'), 'validation_cardinality_error': tensor(4., device='cuda:0')}
|
| 54 |
+
{'training_loss': tensor(2.3822, device='cuda:0'), 'train_loss_ce': tensor(1.1789, device='cuda:0'), 'train_loss_bbox': tensor(0.1039, device='cuda:0'), 'train_loss_giou': tensor(0.3419, device='cuda:0'), 'train_cardinality_error': tensor(28., device='cuda:0'), 'validation_loss': tensor(1.5449, device='cuda:0'), 'validation_loss_ce': tensor(0.5027, device='cuda:0'), 'validation_loss_bbox': tensor(0.0720, device='cuda:0'), 'validation_loss_giou': tensor(0.3411, device='cuda:0'), 'validation_cardinality_error': tensor(3.5385, device='cuda:0')}
|
| 55 |
+
{'training_loss': tensor(0.6928, device='cuda:0'), 'train_loss_ce': tensor(0.3421, device='cuda:0'), 'train_loss_bbox': tensor(0.0502, device='cuda:0'), 'train_loss_giou': tensor(0.0498, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.7679, device='cuda:0'), 'validation_loss_ce': tensor(0.4698, device='cuda:0'), 'validation_loss_bbox': tensor(0.0951, device='cuda:0'), 'validation_loss_giou': tensor(0.4114, device='cuda:0'), 'validation_cardinality_error': tensor(3.3077, device='cuda:0')}
|
| 56 |
+
{'training_loss': tensor(0.4587, device='cuda:0'), 'train_loss_ce': tensor(0.2548, device='cuda:0'), 'train_loss_bbox': tensor(0.0098, device='cuda:0'), 'train_loss_giou': tensor(0.0775, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.1728, device='cuda:0'), 'validation_loss_ce': tensor(0.4435, device='cuda:0'), 'validation_loss_bbox': tensor(0.0387, device='cuda:0'), 'validation_loss_giou': tensor(0.2678, device='cuda:0'), 'validation_cardinality_error': tensor(5.7692, device='cuda:0')}
|
| 57 |
+
{'training_loss': tensor(0.9540, device='cuda:0'), 'train_loss_ce': tensor(0.4630, device='cuda:0'), 'train_loss_bbox': tensor(0.0313, device='cuda:0'), 'train_loss_giou': tensor(0.1673, device='cuda:0'), 'train_cardinality_error': tensor(19., device='cuda:0'), 'validation_loss': tensor(1.0820, device='cuda:0'), 'validation_loss_ce': tensor(0.4409, device='cuda:0'), 'validation_loss_bbox': tensor(0.0341, device='cuda:0'), 'validation_loss_giou': tensor(0.2353, device='cuda:0'), 'validation_cardinality_error': tensor(7.6154, device='cuda:0')}
|
| 58 |
+
{'training_loss': tensor(0.6391, device='cuda:0'), 'train_loss_ce': tensor(0.3858, device='cuda:0'), 'train_loss_bbox': tensor(0.0238, device='cuda:0'), 'train_loss_giou': tensor(0.0672, device='cuda:0'), 'train_cardinality_error': tensor(5., device='cuda:0'), 'validation_loss': tensor(1.0656, device='cuda:0'), 'validation_loss_ce': tensor(0.4144, device='cuda:0'), 'validation_loss_bbox': tensor(0.0261, device='cuda:0'), 'validation_loss_giou': tensor(0.2604, device='cuda:0'), 'validation_cardinality_error': tensor(10.8462, device='cuda:0')}
|
| 59 |
+
{'training_loss': tensor(1.7558, device='cuda:0'), 'train_loss_ce': tensor(0.3972, device='cuda:0'), 'train_loss_bbox': tensor(0.1433, device='cuda:0'), 'train_loss_giou': tensor(0.3210, device='cuda:0'), 'train_cardinality_error': tensor(5., device='cuda:0'), 'validation_loss': tensor(1.9132, device='cuda:0'), 'validation_loss_ce': tensor(0.4682, device='cuda:0'), 'validation_loss_bbox': tensor(0.0978, device='cuda:0'), 'validation_loss_giou': tensor(0.4779, device='cuda:0'), 'validation_cardinality_error': tensor(12., device='cuda:0')}
|
| 60 |
+
{'training_loss': tensor(3.0507, device='cuda:0'), 'train_loss_ce': tensor(0.7805, device='cuda:0'), 'train_loss_bbox': tensor(0.1745, device='cuda:0'), 'train_loss_giou': tensor(0.6989, device='cuda:0'), 'train_cardinality_error': tensor(16., device='cuda:0'), 'validation_loss': tensor(2.6096, device='cuda:0'), 'validation_loss_ce': tensor(0.4846, device='cuda:0'), 'validation_loss_bbox': tensor(0.1573, device='cuda:0'), 'validation_loss_giou': tensor(0.6693, device='cuda:0'), 'validation_cardinality_error': tensor(5.8462, device='cuda:0')}
|
| 61 |
+
{'training_loss': tensor(1.7916, device='cuda:0'), 'train_loss_ce': tensor(0.4071, device='cuda:0'), 'train_loss_bbox': tensor(0.1937, device='cuda:0'), 'train_loss_giou': tensor(0.2079, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(3.2897, device='cuda:0'), 'validation_loss_ce': tensor(0.5079, device='cuda:0'), 'validation_loss_bbox': tensor(0.2703, device='cuda:0'), 'validation_loss_giou': tensor(0.7151, device='cuda:0'), 'validation_cardinality_error': tensor(3.2308, device='cuda:0')}
|
| 62 |
+
{'training_loss': tensor(3.2579, device='cuda:0'), 'train_loss_ce': tensor(0.7360, device='cuda:0'), 'train_loss_bbox': tensor(0.2364, device='cuda:0'), 'train_loss_giou': tensor(0.6700, device='cuda:0'), 'train_cardinality_error': tensor(4., device='cuda:0'), 'validation_loss': tensor(2.0824, device='cuda:0'), 'validation_loss_ce': tensor(0.5116, device='cuda:0'), 'validation_loss_bbox': tensor(0.1048, device='cuda:0'), 'validation_loss_giou': tensor(0.5234, device='cuda:0'), 'validation_cardinality_error': tensor(6.9231, device='cuda:0')}
|
| 63 |
+
{'training_loss': tensor(3.7396, device='cuda:0'), 'train_loss_ce': tensor(0.4191, device='cuda:0'), 'train_loss_bbox': tensor(0.3937, device='cuda:0'), 'train_loss_giou': tensor(0.6760, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(3.1265, device='cuda:0'), 'validation_loss_ce': tensor(0.5484, device='cuda:0'), 'validation_loss_bbox': tensor(0.2265, device='cuda:0'), 'validation_loss_giou': tensor(0.7228, device='cuda:0'), 'validation_cardinality_error': tensor(4.3077, device='cuda:0')}
|
| 64 |
+
{'training_loss': tensor(2.6312, device='cuda:0'), 'train_loss_ce': tensor(0.3234, device='cuda:0'), 'train_loss_bbox': tensor(0.2922, device='cuda:0'), 'train_loss_giou': tensor(0.4235, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(3.3476, device='cuda:0'), 'validation_loss_ce': tensor(0.5086, device='cuda:0'), 'validation_loss_bbox': tensor(0.2609, device='cuda:0'), 'validation_loss_giou': tensor(0.7671, device='cuda:0'), 'validation_cardinality_error': tensor(3.8462, device='cuda:0')}
|
| 65 |
+
{'training_loss': tensor(1.7406, device='cuda:0'), 'train_loss_ce': tensor(0.5410, device='cuda:0'), 'train_loss_bbox': tensor(0.0662, device='cuda:0'), 'train_loss_giou': tensor(0.4344, device='cuda:0'), 'train_cardinality_error': tensor(3., device='cuda:0'), 'validation_loss': tensor(3.2995, device='cuda:0'), 'validation_loss_ce': tensor(0.5209, device='cuda:0'), 'validation_loss_bbox': tensor(0.2275, device='cuda:0'), 'validation_loss_giou': tensor(0.8205, device='cuda:0'), 'validation_cardinality_error': tensor(6.3846, device='cuda:0')}
|
| 66 |
+
{'training_loss': tensor(2.6756, device='cuda:0'), 'train_loss_ce': tensor(0.4669, device='cuda:0'), 'train_loss_bbox': tensor(0.2744, device='cuda:0'), 'train_loss_giou': tensor(0.4184, device='cuda:0'), 'train_cardinality_error': tensor(2., device='cuda:0'), 'validation_loss': tensor(3.9060, device='cuda:0'), 'validation_loss_ce': tensor(0.5255, device='cuda:0'), 'validation_loss_bbox': tensor(0.3109, device='cuda:0'), 'validation_loss_giou': tensor(0.9129, device='cuda:0'), 'validation_cardinality_error': tensor(7.3846, device='cuda:0')}
|
| 67 |
+
{'training_loss': tensor(2.6693, device='cuda:0'), 'train_loss_ce': tensor(0.5077, device='cuda:0'), 'train_loss_bbox': tensor(0.1733, device='cuda:0'), 'train_loss_giou': tensor(0.6475, device='cuda:0'), 'train_cardinality_error': tensor(2., device='cuda:0'), 'validation_loss': tensor(3.3320, device='cuda:0'), 'validation_loss_ce': tensor(0.5506, device='cuda:0'), 'validation_loss_bbox': tensor(0.2695, device='cuda:0'), 'validation_loss_giou': tensor(0.7171, device='cuda:0'), 'validation_cardinality_error': tensor(6., device='cuda:0')}
|
| 68 |
+
{'training_loss': tensor(3.8867, device='cuda:0'), 'train_loss_ce': tensor(0.6607, device='cuda:0'), 'train_loss_bbox': tensor(0.3213, device='cuda:0'), 'train_loss_giou': tensor(0.8097, device='cuda:0'), 'train_cardinality_error': tensor(4., device='cuda:0'), 'validation_loss': tensor(3.0807, device='cuda:0'), 'validation_loss_ce': tensor(0.5218, device='cuda:0'), 'validation_loss_bbox': tensor(0.2218, device='cuda:0'), 'validation_loss_giou': tensor(0.7249, device='cuda:0'), 'validation_cardinality_error': tensor(3.8462, device='cuda:0')}
|
| 69 |
+
{'training_loss': tensor(2.8666, device='cuda:0'), 'train_loss_ce': tensor(0.4696, device='cuda:0'), 'train_loss_bbox': tensor(0.2954, device='cuda:0'), 'train_loss_giou': tensor(0.4601, device='cuda:0'), 'train_cardinality_error': tensor(2., device='cuda:0'), 'validation_loss': tensor(3.0620, device='cuda:0'), 'validation_loss_ce': tensor(0.5262, device='cuda:0'), 'validation_loss_bbox': tensor(0.2149, device='cuda:0'), 'validation_loss_giou': tensor(0.7306, device='cuda:0'), 'validation_cardinality_error': tensor(3.8462, device='cuda:0')}
|
| 70 |
+
{'training_loss': tensor(2.1570, device='cuda:0'), 'train_loss_ce': tensor(0.6110, device='cuda:0'), 'train_loss_bbox': tensor(0.0905, device='cuda:0'), 'train_loss_giou': tensor(0.5468, device='cuda:0'), 'train_cardinality_error': tensor(7., device='cuda:0'), 'validation_loss': tensor(2.8941, device='cuda:0'), 'validation_loss_ce': tensor(0.5307, device='cuda:0'), 'validation_loss_bbox': tensor(0.2003, device='cuda:0'), 'validation_loss_giou': tensor(0.6809, device='cuda:0'), 'validation_cardinality_error': tensor(3.8462, device='cuda:0')}
|
| 71 |
+
{'training_loss': tensor(4.1725, device='cuda:0'), 'train_loss_ce': tensor(0.9266, device='cuda:0'), 'train_loss_bbox': tensor(0.2818, device='cuda:0'), 'train_loss_giou': tensor(0.9185, device='cuda:0'), 'train_cardinality_error': tensor(11., device='cuda:0'), 'validation_loss': tensor(3.3124, device='cuda:0'), 'validation_loss_ce': tensor(0.5442, device='cuda:0'), 'validation_loss_bbox': tensor(0.2570, device='cuda:0'), 'validation_loss_giou': tensor(0.7416, device='cuda:0'), 'validation_cardinality_error': tensor(3.8462, device='cuda:0')}
|
| 72 |
+
{'training_loss': tensor(1.4272, device='cuda:0'), 'train_loss_ce': tensor(0.3878, device='cuda:0'), 'train_loss_bbox': tensor(0.0514, device='cuda:0'), 'train_loss_giou': tensor(0.3911, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.9401, device='cuda:0'), 'validation_loss_ce': tensor(0.5416, device='cuda:0'), 'validation_loss_bbox': tensor(0.2045, device='cuda:0'), 'validation_loss_giou': tensor(0.6879, device='cuda:0'), 'validation_cardinality_error': tensor(3.8462, device='cuda:0')}
|
| 73 |
```
|
| 74 |
|
| 75 |
## Examples
|