npvinHnivqn commited on
Commit
90516c9
1 Parent(s): 9d7a413

Update README file

Browse files
Files changed (1) hide show
  1. README.md +43 -43
README.md CHANGED
@@ -13,34 +13,34 @@ IoU metric: bbox
13
  Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
14
  Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.000
15
  Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.000
16
- Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.000
17
- Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.000
18
  Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000
19
  Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
20
- Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.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.053
27
- Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.096
28
- Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.063
29
  Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000
30
  Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
31
- Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.053
32
- Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.100
33
- Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.179
34
- Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.211
35
  Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000
36
  Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
37
- Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.232
38
  ```
39
 
40
  ## Config
41
  - dataset: NIH
42
  - original model: hustvl/yolos-tiny
43
- - lr: 0.0001
44
  - dropout_rate: 0.1
45
  - weight_decay: 0.05
46
  - max_epochs: 30
@@ -49,37 +49,37 @@ IoU metric: bbox
49
  ## Logging
50
  ### Training process
51
  ```
52
- {'validation_loss': tensor(6.9559, device='cuda:0'), 'validation_loss_ce': tensor(2.4853, device='cuda:0'), 'validation_loss_bbox': tensor(0.5189, device='cuda:0'), 'validation_loss_giou': tensor(0.9380, device='cuda:0'), 'validation_cardinality_error': tensor(99., device='cuda:0')}
53
- {'training_loss': tensor(2.5294, device='cuda:0'), 'train_loss_ce': tensor(0.5418, device='cuda:0'), 'train_loss_bbox': tensor(0.1860, device='cuda:0'), 'train_loss_giou': tensor(0.5287, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.5128, device='cuda:0'), 'validation_loss_ce': tensor(0.4620, device='cuda:0'), 'validation_loss_bbox': tensor(0.1888, device='cuda:0'), 'validation_loss_giou': tensor(0.5533, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
54
- {'training_loss': tensor(2.2589, device='cuda:0'), 'train_loss_ce': tensor(0.4520, device='cuda:0'), 'train_loss_bbox': tensor(0.1275, device='cuda:0'), 'train_loss_giou': tensor(0.5847, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.1865, device='cuda:0'), 'validation_loss_ce': tensor(0.4225, device='cuda:0'), 'validation_loss_bbox': tensor(0.1510, device='cuda:0'), 'validation_loss_giou': tensor(0.5045, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
55
- {'training_loss': tensor(2.2895, device='cuda:0'), 'train_loss_ce': tensor(0.5192, device='cuda:0'), 'train_loss_bbox': tensor(0.1477, device='cuda:0'), 'train_loss_giou': tensor(0.5158, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.1751, device='cuda:0'), 'validation_loss_ce': tensor(0.4340, device='cuda:0'), 'validation_loss_bbox': tensor(0.1528, device='cuda:0'), 'validation_loss_giou': tensor(0.4887, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
56
- {'training_loss': tensor(2.1921, device='cuda:0'), 'train_loss_ce': tensor(0.4626, device='cuda:0'), 'train_loss_bbox': tensor(0.1593, device='cuda:0'), 'train_loss_giou': tensor(0.4666, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.3439, device='cuda:0'), 'validation_loss_ce': tensor(0.4168, device='cuda:0'), 'validation_loss_bbox': tensor(0.1764, device='cuda:0'), 'validation_loss_giou': tensor(0.5226, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
57
- {'training_loss': tensor(1.6664, device='cuda:0'), 'train_loss_ce': tensor(0.4768, device='cuda:0'), 'train_loss_bbox': tensor(0.0885, device='cuda:0'), 'train_loss_giou': tensor(0.3736, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.1216, device='cuda:0'), 'validation_loss_ce': tensor(0.4001, device='cuda:0'), 'validation_loss_bbox': tensor(0.1456, device='cuda:0'), 'validation_loss_giou': tensor(0.4969, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
58
- {'training_loss': tensor(2.5535, device='cuda:0'), 'train_loss_ce': tensor(0.4144, device='cuda:0'), 'train_loss_bbox': tensor(0.1866, device='cuda:0'), 'train_loss_giou': tensor(0.6031, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.3127, device='cuda:0'), 'validation_loss_ce': tensor(0.3946, device='cuda:0'), 'validation_loss_bbox': tensor(0.1638, device='cuda:0'), 'validation_loss_giou': tensor(0.5496, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
59
- {'training_loss': tensor(2.1711, device='cuda:0'), 'train_loss_ce': tensor(0.4254, device='cuda:0'), 'train_loss_bbox': tensor(0.1532, device='cuda:0'), 'train_loss_giou': tensor(0.4897, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.0774, device='cuda:0'), 'validation_loss_ce': tensor(0.4161, device='cuda:0'), 'validation_loss_bbox': tensor(0.1417, device='cuda:0'), 'validation_loss_giou': tensor(0.4764, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
60
- {'training_loss': tensor(1.7319, device='cuda:0'), 'train_loss_ce': tensor(0.4199, device='cuda:0'), 'train_loss_bbox': tensor(0.0986, device='cuda:0'), 'train_loss_giou': tensor(0.4096, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.0720, device='cuda:0'), 'validation_loss_ce': tensor(0.3979, device='cuda:0'), 'validation_loss_bbox': tensor(0.1357, device='cuda:0'), 'validation_loss_giou': tensor(0.4978, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
61
- {'training_loss': tensor(1.5806, device='cuda:0'), 'train_loss_ce': tensor(0.3604, device='cuda:0'), 'train_loss_bbox': tensor(0.1325, device='cuda:0'), 'train_loss_giou': tensor(0.2790, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9735, device='cuda:0'), 'validation_loss_ce': tensor(0.3799, device='cuda:0'), 'validation_loss_bbox': tensor(0.1335, device='cuda:0'), 'validation_loss_giou': tensor(0.4631, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
62
- {'training_loss': tensor(2.0937, device='cuda:0'), 'train_loss_ce': tensor(0.4021, device='cuda:0'), 'train_loss_bbox': tensor(0.1257, device='cuda:0'), 'train_loss_giou': tensor(0.5315, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9569, device='cuda:0'), 'validation_loss_ce': tensor(0.3864, device='cuda:0'), 'validation_loss_bbox': tensor(0.1335, device='cuda:0'), 'validation_loss_giou': tensor(0.4515, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
63
- {'training_loss': tensor(1.1840, device='cuda:0'), 'train_loss_ce': tensor(0.3635, device='cuda:0'), 'train_loss_bbox': tensor(0.0550, device='cuda:0'), 'train_loss_giou': tensor(0.2729, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9329, device='cuda:0'), 'validation_loss_ce': tensor(0.3740, device='cuda:0'), 'validation_loss_bbox': tensor(0.1274, device='cuda:0'), 'validation_loss_giou': tensor(0.4610, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
64
- {'training_loss': tensor(2.0942, device='cuda:0'), 'train_loss_ce': tensor(0.3967, device='cuda:0'), 'train_loss_bbox': tensor(0.1509, device='cuda:0'), 'train_loss_giou': tensor(0.4715, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9473, device='cuda:0'), 'validation_loss_ce': tensor(0.3690, device='cuda:0'), 'validation_loss_bbox': tensor(0.1283, device='cuda:0'), 'validation_loss_giou': tensor(0.4683, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
65
- {'training_loss': tensor(1.6510, device='cuda:0'), 'train_loss_ce': tensor(0.2956, device='cuda:0'), 'train_loss_bbox': tensor(0.0785, device='cuda:0'), 'train_loss_giou': tensor(0.4814, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.8971, device='cuda:0'), 'validation_loss_ce': tensor(0.3623, device='cuda:0'), 'validation_loss_bbox': tensor(0.1229, device='cuda:0'), 'validation_loss_giou': tensor(0.4603, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
66
- {'training_loss': tensor(1.3376, device='cuda:0'), 'train_loss_ce': tensor(0.2663, device='cuda:0'), 'train_loss_bbox': tensor(0.0985, device='cuda:0'), 'train_loss_giou': tensor(0.2893, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.8048, device='cuda:0'), 'validation_loss_ce': tensor(0.3550, device='cuda:0'), 'validation_loss_bbox': tensor(0.1180, device='cuda:0'), 'validation_loss_giou': tensor(0.4298, device='cuda:0'), 'validation_cardinality_error': tensor(0.9899, device='cuda:0')}
67
- {'training_loss': tensor(1.8508, device='cuda:0'), 'train_loss_ce': tensor(0.4634, device='cuda:0'), 'train_loss_bbox': tensor(0.1160, device='cuda:0'), 'train_loss_giou': tensor(0.4036, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.8110, device='cuda:0'), 'validation_loss_ce': tensor(0.3658, device='cuda:0'), 'validation_loss_bbox': tensor(0.1133, device='cuda:0'), 'validation_loss_giou': tensor(0.4393, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
68
- {'training_loss': tensor(2.2132, device='cuda:0'), 'train_loss_ce': tensor(0.3865, device='cuda:0'), 'train_loss_bbox': tensor(0.1733, device='cuda:0'), 'train_loss_giou': tensor(0.4802, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.8956, device='cuda:0'), 'validation_loss_ce': tensor(0.3703, device='cuda:0'), 'validation_loss_bbox': tensor(0.1241, device='cuda:0'), 'validation_loss_giou': tensor(0.4525, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
69
- {'training_loss': tensor(1.4466, device='cuda:0'), 'train_loss_ce': tensor(0.4195, device='cuda:0'), 'train_loss_bbox': tensor(0.0736, device='cuda:0'), 'train_loss_giou': tensor(0.3295, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9727, device='cuda:0'), 'validation_loss_ce': tensor(0.3733, device='cuda:0'), 'validation_loss_bbox': tensor(0.1318, device='cuda:0'), 'validation_loss_giou': tensor(0.4702, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
70
- {'training_loss': tensor(1.6351, device='cuda:0'), 'train_loss_ce': tensor(0.3461, device='cuda:0'), 'train_loss_bbox': tensor(0.0929, device='cuda:0'), 'train_loss_giou': tensor(0.4122, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.8800, device='cuda:0'), 'validation_loss_ce': tensor(0.3603, device='cuda:0'), 'validation_loss_bbox': tensor(0.1224, device='cuda:0'), 'validation_loss_giou': tensor(0.4539, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
71
- {'training_loss': tensor(2.1954, device='cuda:0'), 'train_loss_ce': tensor(0.3990, device='cuda:0'), 'train_loss_bbox': tensor(0.1613, device='cuda:0'), 'train_loss_giou': tensor(0.4950, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.8715, device='cuda:0'), 'validation_loss_ce': tensor(0.3536, device='cuda:0'), 'validation_loss_bbox': tensor(0.1234, device='cuda:0'), 'validation_loss_giou': tensor(0.4504, device='cuda:0'), 'validation_cardinality_error': tensor(0.9192, device='cuda:0')}
72
- {'training_loss': tensor(1.8643, device='cuda:0'), 'train_loss_ce': tensor(0.3877, device='cuda:0'), 'train_loss_bbox': tensor(0.0971, device='cuda:0'), 'train_loss_giou': tensor(0.4955, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.7371, device='cuda:0'), 'validation_loss_ce': tensor(0.3362, device='cuda:0'), 'validation_loss_bbox': tensor(0.1148, device='cuda:0'), 'validation_loss_giou': tensor(0.4134, device='cuda:0'), 'validation_cardinality_error': tensor(0.9394, device='cuda:0')}
73
- {'training_loss': tensor(1.3830, device='cuda:0'), 'train_loss_ce': tensor(0.1839, device='cuda:0'), 'train_loss_bbox': tensor(0.0825, device='cuda:0'), 'train_loss_giou': tensor(0.3932, device='cuda:0'), 'train_cardinality_error': tensor(0.6000, device='cuda:0'), 'validation_loss': tensor(1.8157, device='cuda:0'), 'validation_loss_ce': tensor(0.3523, device='cuda:0'), 'validation_loss_bbox': tensor(0.1163, device='cuda:0'), 'validation_loss_giou': tensor(0.4410, device='cuda:0'), 'validation_cardinality_error': tensor(0.9293, device='cuda:0')}
74
- {'training_loss': tensor(1.6100, device='cuda:0'), 'train_loss_ce': tensor(0.3392, device='cuda:0'), 'train_loss_bbox': tensor(0.1034, device='cuda:0'), 'train_loss_giou': tensor(0.3770, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.7169, device='cuda:0'), 'validation_loss_ce': tensor(0.3482, device='cuda:0'), 'validation_loss_bbox': tensor(0.1110, device='cuda:0'), 'validation_loss_giou': tensor(0.4070, device='cuda:0'), 'validation_cardinality_error': tensor(0.9192, device='cuda:0')}
75
- {'training_loss': tensor(1.5429, device='cuda:0'), 'train_loss_ce': tensor(0.3210, device='cuda:0'), 'train_loss_bbox': tensor(0.0945, device='cuda:0'), 'train_loss_giou': tensor(0.3747, device='cuda:0'), 'train_cardinality_error': tensor(0.8000, device='cuda:0'), 'validation_loss': tensor(1.7941, device='cuda:0'), 'validation_loss_ce': tensor(0.3513, device='cuda:0'), 'validation_loss_bbox': tensor(0.1161, device='cuda:0'), 'validation_loss_giou': tensor(0.4312, device='cuda:0'), 'validation_cardinality_error': tensor(0.8990, device='cuda:0')}
76
- {'training_loss': tensor(1.4975, device='cuda:0'), 'train_loss_ce': tensor(0.3537, device='cuda:0'), 'train_loss_bbox': tensor(0.0965, device='cuda:0'), 'train_loss_giou': tensor(0.3307, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.7279, device='cuda:0'), 'validation_loss_ce': tensor(0.3444, device='cuda:0'), 'validation_loss_bbox': tensor(0.1122, device='cuda:0'), 'validation_loss_giou': tensor(0.4112, device='cuda:0'), 'validation_cardinality_error': tensor(0.8081, device='cuda:0')}
77
- {'training_loss': tensor(1.6512, device='cuda:0'), 'train_loss_ce': tensor(0.3698, device='cuda:0'), 'train_loss_bbox': tensor(0.1170, device='cuda:0'), 'train_loss_giou': tensor(0.3481, device='cuda:0'), 'train_cardinality_error': tensor(0.4000, device='cuda:0'), 'validation_loss': tensor(1.8547, device='cuda:0'), 'validation_loss_ce': tensor(0.3486, device='cuda:0'), 'validation_loss_bbox': tensor(0.1224, device='cuda:0'), 'validation_loss_giou': tensor(0.4470, device='cuda:0'), 'validation_cardinality_error': tensor(0.7576, device='cuda:0')}
78
- {'training_loss': tensor(2.2053, device='cuda:0'), 'train_loss_ce': tensor(0.3561, device='cuda:0'), 'train_loss_bbox': tensor(0.1675, device='cuda:0'), 'train_loss_giou': tensor(0.5059, device='cuda:0'), 'train_cardinality_error': tensor(0.8000, device='cuda:0'), 'validation_loss': tensor(1.8177, device='cuda:0'), 'validation_loss_ce': tensor(0.3480, device='cuda:0'), 'validation_loss_bbox': tensor(0.1206, device='cuda:0'), 'validation_loss_giou': tensor(0.4333, device='cuda:0'), 'validation_cardinality_error': tensor(0.7071, device='cuda:0')}
79
- {'training_loss': tensor(1.6742, device='cuda:0'), 'train_loss_ce': tensor(0.3054, device='cuda:0'), 'train_loss_bbox': tensor(0.1093, device='cuda:0'), 'train_loss_giou': tensor(0.4112, device='cuda:0'), 'train_cardinality_error': tensor(0.8000, device='cuda:0'), 'validation_loss': tensor(1.7886, device='cuda:0'), 'validation_loss_ce': tensor(0.3294, device='cuda:0'), 'validation_loss_bbox': tensor(0.1209, device='cuda:0'), 'validation_loss_giou': tensor(0.4274, device='cuda:0'), 'validation_cardinality_error': tensor(0.6364, device='cuda:0')}
80
- {'training_loss': tensor(0.8524, device='cuda:0'), 'train_loss_ce': tensor(0.1265, device='cuda:0'), 'train_loss_bbox': tensor(0.0692, device='cuda:0'), 'train_loss_giou': tensor(0.1898, device='cuda:0'), 'train_cardinality_error': tensor(0.8000, device='cuda:0'), 'validation_loss': tensor(1.7781, device='cuda:0'), 'validation_loss_ce': tensor(0.3431, device='cuda:0'), 'validation_loss_bbox': tensor(0.1189, device='cuda:0'), 'validation_loss_giou': tensor(0.4203, device='cuda:0'), 'validation_cardinality_error': tensor(0.6869, device='cuda:0')}
81
- {'training_loss': tensor(1.4292, device='cuda:0'), 'train_loss_ce': tensor(0.3210, device='cuda:0'), 'train_loss_bbox': tensor(0.0939, device='cuda:0'), 'train_loss_giou': tensor(0.3193, device='cuda:0'), 'train_cardinality_error': tensor(0.6000, device='cuda:0'), 'validation_loss': tensor(1.8680, device='cuda:0'), 'validation_loss_ce': tensor(0.3525, device='cuda:0'), 'validation_loss_bbox': tensor(0.1307, device='cuda:0'), 'validation_loss_giou': tensor(0.4311, device='cuda:0'), 'validation_cardinality_error': tensor(0.8788, device='cuda:0')}
82
- {'training_loss': tensor(1.2447, device='cuda:0'), 'train_loss_ce': tensor(0.2517, device='cuda:0'), 'train_loss_bbox': tensor(0.0609, device='cuda:0'), 'train_loss_giou': tensor(0.3442, device='cuda:0'), 'train_cardinality_error': tensor(0.6000, device='cuda:0'), 'validation_loss': tensor(1.7613, device='cuda:0'), 'validation_loss_ce': tensor(0.3511, device='cuda:0'), 'validation_loss_bbox': tensor(0.1156, device='cuda:0'), 'validation_loss_giou': tensor(0.4160, device='cuda:0'), 'validation_cardinality_error': tensor(0.5455, device='cuda:0')}
83
  ```
84
 
85
  ## Examples
 
13
  Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
14
  Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.000
15
  Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.000
16
+ Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.003
17
+ Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.008
18
  Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000
19
  Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
20
+ Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.008
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.012
27
+ Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.023
28
+ Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.011
29
  Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000
30
  Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
31
+ Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.012
32
+ Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.056
33
+ Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.118
34
+ Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.145
35
  Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000
36
  Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
37
+ Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.146
38
  ```
39
 
40
  ## Config
41
  - dataset: NIH
42
  - original model: hustvl/yolos-tiny
43
+ - lr: 1e-05
44
  - dropout_rate: 0.1
45
  - weight_decay: 0.05
46
  - max_epochs: 30
 
49
  ## Logging
50
  ### Training process
51
  ```
52
+ {'validation_loss': tensor(6.7207, device='cuda:0'), 'validation_loss_ce': tensor(2.1866, device='cuda:0'), 'validation_loss_bbox': tensor(0.5249, device='cuda:0'), 'validation_loss_giou': tensor(0.9547, device='cuda:0'), 'validation_cardinality_error': tensor(98.5312, device='cuda:0')}
53
+ {'training_loss': tensor(3.7287, device='cuda:0'), 'train_loss_ce': tensor(1.3098, device='cuda:0'), 'train_loss_bbox': tensor(0.2119, device='cuda:0'), 'train_loss_giou': tensor(0.6797, device='cuda:0'), 'train_cardinality_error': tensor(0.8000, device='cuda:0'), 'validation_loss': tensor(3.4332, device='cuda:0'), 'validation_loss_ce': tensor(1.2399, device='cuda:0'), 'validation_loss_bbox': tensor(0.2065, device='cuda:0'), 'validation_loss_giou': tensor(0.5804, device='cuda:0'), 'validation_cardinality_error': tensor(1.0909, device='cuda:0')}
54
+ {'training_loss': tensor(2.8569, device='cuda:0'), 'train_loss_ce': tensor(0.5845, device='cuda:0'), 'train_loss_bbox': tensor(0.2106, device='cuda:0'), 'train_loss_giou': tensor(0.6097, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.4866, device='cuda:0'), 'validation_loss_ce': tensor(0.5491, device='cuda:0'), 'validation_loss_bbox': tensor(0.1759, device='cuda:0'), 'validation_loss_giou': tensor(0.5290, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
55
+ {'training_loss': tensor(2.2491, device='cuda:0'), 'train_loss_ce': tensor(0.5182, device='cuda:0'), 'train_loss_bbox': tensor(0.1662, device='cuda:0'), 'train_loss_giou': tensor(0.4500, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.3079, device='cuda:0'), 'validation_loss_ce': tensor(0.4791, device='cuda:0'), 'validation_loss_bbox': tensor(0.1619, device='cuda:0'), 'validation_loss_giou': tensor(0.5096, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
56
+ {'training_loss': tensor(1.9377, device='cuda:0'), 'train_loss_ce': tensor(0.5210, device='cuda:0'), 'train_loss_bbox': tensor(0.1273, device='cuda:0'), 'train_loss_giou': tensor(0.3902, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.2293, device='cuda:0'), 'validation_loss_ce': tensor(0.4691, device='cuda:0'), 'validation_loss_bbox': tensor(0.1544, device='cuda:0'), 'validation_loss_giou': tensor(0.4940, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
57
+ {'training_loss': tensor(2.1924, device='cuda:0'), 'train_loss_ce': tensor(0.5255, device='cuda:0'), 'train_loss_bbox': tensor(0.1414, device='cuda:0'), 'train_loss_giou': tensor(0.4800, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.1471, device='cuda:0'), 'validation_loss_ce': tensor(0.4648, device='cuda:0'), 'validation_loss_bbox': tensor(0.1457, device='cuda:0'), 'validation_loss_giou': tensor(0.4770, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
58
+ {'training_loss': tensor(2.4833, device='cuda:0'), 'train_loss_ce': tensor(0.4674, device='cuda:0'), 'train_loss_bbox': tensor(0.2011, device='cuda:0'), 'train_loss_giou': tensor(0.5053, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.1557, device='cuda:0'), 'validation_loss_ce': tensor(0.4630, device='cuda:0'), 'validation_loss_bbox': tensor(0.1464, device='cuda:0'), 'validation_loss_giou': tensor(0.4804, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
59
+ {'training_loss': tensor(1.9442, device='cuda:0'), 'train_loss_ce': tensor(0.4237, device='cuda:0'), 'train_loss_bbox': tensor(0.1272, device='cuda:0'), 'train_loss_giou': tensor(0.4424, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.1246, device='cuda:0'), 'validation_loss_ce': tensor(0.4547, device='cuda:0'), 'validation_loss_bbox': tensor(0.1406, device='cuda:0'), 'validation_loss_giou': tensor(0.4833, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
60
+ {'training_loss': tensor(1.8449, device='cuda:0'), 'train_loss_ce': tensor(0.5204, device='cuda:0'), 'train_loss_bbox': tensor(0.1064, device='cuda:0'), 'train_loss_giou': tensor(0.3963, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.0914, device='cuda:0'), 'validation_loss_ce': tensor(0.4524, device='cuda:0'), 'validation_loss_bbox': tensor(0.1409, device='cuda:0'), 'validation_loss_giou': tensor(0.4673, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
61
+ {'training_loss': tensor(2.2881, device='cuda:0'), 'train_loss_ce': tensor(0.4765, device='cuda:0'), 'train_loss_bbox': tensor(0.1549, device='cuda:0'), 'train_loss_giou': tensor(0.5186, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.1367, device='cuda:0'), 'validation_loss_ce': tensor(0.4505, device='cuda:0'), 'validation_loss_bbox': tensor(0.1434, device='cuda:0'), 'validation_loss_giou': tensor(0.4846, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
62
+ {'training_loss': tensor(1.5250, device='cuda:0'), 'train_loss_ce': tensor(0.4858, device='cuda:0'), 'train_loss_bbox': tensor(0.0681, device='cuda:0'), 'train_loss_giou': tensor(0.3494, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.0727, device='cuda:0'), 'validation_loss_ce': tensor(0.4480, device='cuda:0'), 'validation_loss_bbox': tensor(0.1342, device='cuda:0'), 'validation_loss_giou': tensor(0.4769, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
63
+ {'training_loss': tensor(1.5382, device='cuda:0'), 'train_loss_ce': tensor(0.3929, device='cuda:0'), 'train_loss_bbox': tensor(0.1066, device='cuda:0'), 'train_loss_giou': tensor(0.3061, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9921, device='cuda:0'), 'validation_loss_ce': tensor(0.4464, device='cuda:0'), 'validation_loss_bbox': tensor(0.1298, device='cuda:0'), 'validation_loss_giou': tensor(0.4483, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
64
+ {'training_loss': tensor(1.8638, device='cuda:0'), 'train_loss_ce': tensor(0.4090, device='cuda:0'), 'train_loss_bbox': tensor(0.1044, device='cuda:0'), 'train_loss_giou': tensor(0.4665, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.0358, device='cuda:0'), 'validation_loss_ce': tensor(0.4468, device='cuda:0'), 'validation_loss_bbox': tensor(0.1343, device='cuda:0'), 'validation_loss_giou': tensor(0.4588, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
65
+ {'training_loss': tensor(1.6616, device='cuda:0'), 'train_loss_ce': tensor(0.4866, device='cuda:0'), 'train_loss_bbox': tensor(0.0970, device='cuda:0'), 'train_loss_giou': tensor(0.3450, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.0047, device='cuda:0'), 'validation_loss_ce': tensor(0.4417, device='cuda:0'), 'validation_loss_bbox': tensor(0.1302, device='cuda:0'), 'validation_loss_giou': tensor(0.4559, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
66
+ {'training_loss': tensor(1.4619, device='cuda:0'), 'train_loss_ce': tensor(0.4675, device='cuda:0'), 'train_loss_bbox': tensor(0.1028, device='cuda:0'), 'train_loss_giou': tensor(0.2401, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.0066, device='cuda:0'), 'validation_loss_ce': tensor(0.4412, device='cuda:0'), 'validation_loss_bbox': tensor(0.1329, device='cuda:0'), 'validation_loss_giou': tensor(0.4505, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
67
+ {'training_loss': tensor(1.5520, device='cuda:0'), 'train_loss_ce': tensor(0.4766, device='cuda:0'), 'train_loss_bbox': tensor(0.0950, device='cuda:0'), 'train_loss_giou': tensor(0.3002, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9805, device='cuda:0'), 'validation_loss_ce': tensor(0.4422, device='cuda:0'), 'validation_loss_bbox': tensor(0.1306, device='cuda:0'), 'validation_loss_giou': tensor(0.4428, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
68
+ {'training_loss': tensor(2.0331, device='cuda:0'), 'train_loss_ce': tensor(0.5165, device='cuda:0'), 'train_loss_bbox': tensor(0.1336, device='cuda:0'), 'train_loss_giou': tensor(0.4242, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9927, device='cuda:0'), 'validation_loss_ce': tensor(0.4389, device='cuda:0'), 'validation_loss_bbox': tensor(0.1282, device='cuda:0'), 'validation_loss_giou': tensor(0.4563, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
69
+ {'training_loss': tensor(1.4602, device='cuda:0'), 'train_loss_ce': tensor(0.2981, device='cuda:0'), 'train_loss_bbox': tensor(0.1021, device='cuda:0'), 'train_loss_giou': tensor(0.3257, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9925, device='cuda:0'), 'validation_loss_ce': tensor(0.4381, device='cuda:0'), 'validation_loss_bbox': tensor(0.1300, device='cuda:0'), 'validation_loss_giou': tensor(0.4522, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
70
+ {'training_loss': tensor(1.8886, device='cuda:0'), 'train_loss_ce': tensor(0.4785, device='cuda:0'), 'train_loss_bbox': tensor(0.1082, device='cuda:0'), 'train_loss_giou': tensor(0.4345, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9744, device='cuda:0'), 'validation_loss_ce': tensor(0.4333, device='cuda:0'), 'validation_loss_bbox': tensor(0.1284, device='cuda:0'), 'validation_loss_giou': tensor(0.4496, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
71
+ {'training_loss': tensor(1.8621, device='cuda:0'), 'train_loss_ce': tensor(0.5109, device='cuda:0'), 'train_loss_bbox': tensor(0.0748, device='cuda:0'), 'train_loss_giou': tensor(0.4886, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9745, device='cuda:0'), 'validation_loss_ce': tensor(0.4335, device='cuda:0'), 'validation_loss_bbox': tensor(0.1278, device='cuda:0'), 'validation_loss_giou': tensor(0.4509, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
72
+ {'training_loss': tensor(1.0607, device='cuda:0'), 'train_loss_ce': tensor(0.4695, device='cuda:0'), 'train_loss_bbox': tensor(0.0499, device='cuda:0'), 'train_loss_giou': tensor(0.1708, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9522, device='cuda:0'), 'validation_loss_ce': tensor(0.4317, device='cuda:0'), 'validation_loss_bbox': tensor(0.1251, device='cuda:0'), 'validation_loss_giou': tensor(0.4474, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
73
+ {'training_loss': tensor(1.7454, device='cuda:0'), 'train_loss_ce': tensor(0.4005, device='cuda:0'), 'train_loss_bbox': tensor(0.0675, device='cuda:0'), 'train_loss_giou': tensor(0.5036, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9562, device='cuda:0'), 'validation_loss_ce': tensor(0.4312, device='cuda:0'), 'validation_loss_bbox': tensor(0.1280, device='cuda:0'), 'validation_loss_giou': tensor(0.4424, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
74
+ {'training_loss': tensor(1.1943, device='cuda:0'), 'train_loss_ce': tensor(0.4380, device='cuda:0'), 'train_loss_bbox': tensor(0.0761, device='cuda:0'), 'train_loss_giou': tensor(0.1880, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9790, device='cuda:0'), 'validation_loss_ce': tensor(0.4290, device='cuda:0'), 'validation_loss_bbox': tensor(0.1262, device='cuda:0'), 'validation_loss_giou': tensor(0.4595, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
75
+ {'training_loss': tensor(1.3419, device='cuda:0'), 'train_loss_ce': tensor(0.3899, device='cuda:0'), 'train_loss_bbox': tensor(0.0794, device='cuda:0'), 'train_loss_giou': tensor(0.2774, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9489, device='cuda:0'), 'validation_loss_ce': tensor(0.4249, device='cuda:0'), 'validation_loss_bbox': tensor(0.1241, device='cuda:0'), 'validation_loss_giou': tensor(0.4519, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
76
+ {'training_loss': tensor(1.9394, device='cuda:0'), 'train_loss_ce': tensor(0.3998, device='cuda:0'), 'train_loss_bbox': tensor(0.1388, device='cuda:0'), 'train_loss_giou': tensor(0.4227, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.0161, device='cuda:0'), 'validation_loss_ce': tensor(0.4221, device='cuda:0'), 'validation_loss_bbox': tensor(0.1321, device='cuda:0'), 'validation_loss_giou': tensor(0.4667, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
77
+ {'training_loss': tensor(1.3021, device='cuda:0'), 'train_loss_ce': tensor(0.3494, device='cuda:0'), 'train_loss_bbox': tensor(0.0630, device='cuda:0'), 'train_loss_giou': tensor(0.3188, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.0121, device='cuda:0'), 'validation_loss_ce': tensor(0.4226, device='cuda:0'), 'validation_loss_bbox': tensor(0.1326, device='cuda:0'), 'validation_loss_giou': tensor(0.4634, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
78
+ {'training_loss': tensor(1.4117, device='cuda:0'), 'train_loss_ce': tensor(0.4587, device='cuda:0'), 'train_loss_bbox': tensor(0.0724, device='cuda:0'), 'train_loss_giou': tensor(0.2954, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9528, device='cuda:0'), 'validation_loss_ce': tensor(0.4217, device='cuda:0'), 'validation_loss_bbox': tensor(0.1242, device='cuda:0'), 'validation_loss_giou': tensor(0.4550, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
79
+ {'training_loss': tensor(1.4619, device='cuda:0'), 'train_loss_ce': tensor(0.3838, device='cuda:0'), 'train_loss_bbox': tensor(0.0605, device='cuda:0'), 'train_loss_giou': tensor(0.3878, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(2.0346, device='cuda:0'), 'validation_loss_ce': tensor(0.4194, device='cuda:0'), 'validation_loss_bbox': tensor(0.1323, device='cuda:0'), 'validation_loss_giou': tensor(0.4769, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
80
+ {'training_loss': tensor(1.2125, device='cuda:0'), 'train_loss_ce': tensor(0.3469, device='cuda:0'), 'train_loss_bbox': tensor(0.0725, device='cuda:0'), 'train_loss_giou': tensor(0.2517, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9540, device='cuda:0'), 'validation_loss_ce': tensor(0.4169, device='cuda:0'), 'validation_loss_bbox': tensor(0.1248, device='cuda:0'), 'validation_loss_giou': tensor(0.4566, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
81
+ {'training_loss': tensor(1.1738, device='cuda:0'), 'train_loss_ce': tensor(0.4249, device='cuda:0'), 'train_loss_bbox': tensor(0.0353, device='cuda:0'), 'train_loss_giou': tensor(0.2863, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9534, device='cuda:0'), 'validation_loss_ce': tensor(0.4152, device='cuda:0'), 'validation_loss_bbox': tensor(0.1264, device='cuda:0'), 'validation_loss_giou': tensor(0.4531, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
82
+ {'training_loss': tensor(1.2029, device='cuda:0'), 'train_loss_ce': tensor(0.4904, device='cuda:0'), 'train_loss_bbox': tensor(0.0577, device='cuda:0'), 'train_loss_giou': tensor(0.2119, device='cuda:0'), 'train_cardinality_error': tensor(1., device='cuda:0'), 'validation_loss': tensor(1.9811, device='cuda:0'), 'validation_loss_ce': tensor(0.4117, device='cuda:0'), 'validation_loss_bbox': tensor(0.1291, device='cuda:0'), 'validation_loss_giou': tensor(0.4620, device='cuda:0'), 'validation_cardinality_error': tensor(1., device='cuda:0')}
83
  ```
84
 
85
  ## Examples