This commit is contained in:
PatrickBue 2019-11-27 17:21:04 +00:00 коммит произвёл Young Park
Родитель eac9b8e1c1
Коммит 1dde83a9bd
2 изменённых файлов: 2 добавлений и 272 удалений

Просмотреть файл

@ -73,7 +73,6 @@
" plot_pr_curves,\n",
" PlotSettings,\n",
" plot_counts_curves,\n",
" #plot_detection_vs_ground_truth,\n",
" plot_detections\n",
")\n",
"from utils_cv.detection.model import (\n",

Просмотреть файл

@ -230,278 +230,9 @@
},
{
"cell_type": "code",
"execution_count": 23,
"execution_count": null,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Epoch: [0] [ 0/58] eta: 0:00:10 lr: 0.000093 loss: 0.0169 (0.0169) loss_classifier: 0.0111 (0.0111) loss_box_reg: 0.0029 (0.0029) loss_objectness: 0.0002 (0.0002) loss_rpn_box_reg: 0.0028 (0.0028) time: 0.1875 data: 0.0937 max mem: 1118\n",
"Epoch: [0] [30/58] eta: 0:00:05 lr: 0.002722 loss: 0.0193 (0.0323) loss_classifier: 0.0111 (0.0124) loss_box_reg: 0.0029 (0.0029) loss_objectness: 0.0003 (0.0029) loss_rpn_box_reg: 0.0026 (0.0141) time: 0.2036 data: 0.1084 max mem: 1278\n",
"Epoch: [0] [57/58] eta: 0:00:00 lr: 0.005000 loss: 0.0254 (0.0401) loss_classifier: 0.0096 (0.0114) loss_box_reg: 0.0035 (0.0033) loss_objectness: 0.0004 (0.0030) loss_rpn_box_reg: 0.0043 (0.0224) time: 0.1845 data: 0.0921 max mem: 1278\n",
"Epoch: [0] Total time: 0:00:11 (0.1915 s / it)\n",
"creating index...\n",
"index created!\n",
"Test: [ 0/16] eta: 0:00:01 model_time: 0.0312 (0.0312) evaluator_time: 0.0156 (0.0156) time: 0.0781 data: 0.0313 max mem: 1278\n",
"Test: [15/16] eta: 0:00:00 model_time: 0.0312 (0.0378) evaluator_time: 0.0000 (0.0021) time: 0.0625 data: 0.0227 max mem: 1278\n",
"Test: Total time: 0:00:01 (0.0625 s / it)\n",
"Averaged stats: model_time: 0.0312 (0.0378) evaluator_time: 0.0000 (0.0021)\n",
"Accumulating evaluation results...\n",
"DONE (t=0.02s).\n",
"IoU metric: bbox\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.904\n",
" Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.985\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.904\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.921\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.921\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.921\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.921\n",
"Epoch: [1] [ 0/58] eta: 0:00:10 lr: 0.005000 loss: 0.0638 (0.0638) loss_classifier: 0.0075 (0.0075) loss_box_reg: 0.0011 (0.0011) loss_objectness: 0.0046 (0.0046) loss_rpn_box_reg: 0.0507 (0.0507) time: 0.1875 data: 0.0937 max mem: 1278\n",
"Epoch: [1] [30/58] eta: 0:00:05 lr: 0.005000 loss: 0.0247 (0.0635) loss_classifier: 0.0127 (0.0121) loss_box_reg: 0.0039 (0.0040) loss_objectness: 0.0003 (0.0047) loss_rpn_box_reg: 0.0041 (0.0426) time: 0.1845 data: 0.0929 max mem: 1278\n",
"Epoch: [1] [57/58] eta: 0:00:00 lr: 0.005000 loss: 0.0245 (0.0565) loss_classifier: 0.0084 (0.0115) loss_box_reg: 0.0037 (0.0043) loss_objectness: 0.0003 (0.0033) loss_rpn_box_reg: 0.0039 (0.0373) time: 0.1835 data: 0.0914 max mem: 1278\n",
"Epoch: [1] Total time: 0:00:10 (0.1855 s / it)\n",
"creating index...\n",
"index created!\n",
"Test: [ 0/16] eta: 0:00:00 model_time: 0.0312 (0.0312) evaluator_time: 0.0000 (0.0000) time: 0.0625 data: 0.0313 max mem: 1278\n",
"Test: [15/16] eta: 0:00:00 model_time: 0.0312 (0.0313) evaluator_time: 0.0000 (0.0059) time: 0.0616 data: 0.0244 max mem: 1278\n",
"Test: Total time: 0:00:01 (0.0626 s / it)\n",
"Averaged stats: model_time: 0.0312 (0.0313) evaluator_time: 0.0000 (0.0059)\n",
"Accumulating evaluation results...\n",
"DONE (t=0.03s).\n",
"IoU metric: bbox\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.868\n",
" Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.973\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.868\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.891\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.891\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.891\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.891\n",
"Epoch: [2] [ 0/58] eta: 0:00:10 lr: 0.005000 loss: 0.1115 (0.1115) loss_classifier: 0.0079 (0.0079) loss_box_reg: 0.0015 (0.0015) loss_objectness: 0.0100 (0.0100) loss_rpn_box_reg: 0.0921 (0.0921) time: 0.1875 data: 0.0937 max mem: 1278\n",
"Epoch: [2] [30/58] eta: 0:00:05 lr: 0.005000 loss: 0.0269 (0.0414) loss_classifier: 0.0091 (0.0106) loss_box_reg: 0.0027 (0.0039) loss_objectness: 0.0003 (0.0031) loss_rpn_box_reg: 0.0048 (0.0238) time: 0.1829 data: 0.0936 max mem: 1278\n",
"Epoch: [2] [57/58] eta: 0:00:00 lr: 0.005000 loss: 0.0251 (0.0430) loss_classifier: 0.0105 (0.0102) loss_box_reg: 0.0032 (0.0042) loss_objectness: 0.0005 (0.0024) loss_rpn_box_reg: 0.0050 (0.0262) time: 0.1854 data: 0.0920 max mem: 1278\n",
"Epoch: [2] Total time: 0:00:10 (0.1879 s / it)\n",
"creating index...\n",
"index created!\n",
"Test: [ 0/16] eta: 0:00:01 model_time: 0.0212 (0.0212) evaluator_time: 0.0156 (0.0156) time: 0.0757 data: 0.0378 max mem: 1278\n",
"Test: [15/16] eta: 0:00:00 model_time: 0.0312 (0.0326) evaluator_time: 0.0156 (0.0088) time: 0.0623 data: 0.0209 max mem: 1278\n",
"Test: Total time: 0:00:00 (0.0623 s / it)\n",
"Averaged stats: model_time: 0.0312 (0.0326) evaluator_time: 0.0156 (0.0088)\n",
"Accumulating evaluation results...\n",
"DONE (t=0.02s).\n",
"IoU metric: bbox\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.894\n",
" Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.985\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.894\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.912\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.912\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.912\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.912\n",
"Epoch: [3] [ 0/58] eta: 0:00:13 lr: 0.005000 loss: 0.0112 (0.0112) loss_classifier: 0.0060 (0.0060) loss_box_reg: 0.0023 (0.0023) loss_objectness: 0.0003 (0.0003) loss_rpn_box_reg: 0.0025 (0.0025) time: 0.2279 data: 0.1342 max mem: 1278\n",
"Epoch: [3] [30/58] eta: 0:00:05 lr: 0.005000 loss: 0.0179 (0.0373) loss_classifier: 0.0103 (0.0100) loss_box_reg: 0.0032 (0.0036) loss_objectness: 0.0003 (0.0014) loss_rpn_box_reg: 0.0057 (0.0223) time: 0.1842 data: 0.0918 max mem: 1278\n",
"Epoch: [3] [57/58] eta: 0:00:00 lr: 0.005000 loss: 0.0289 (0.0385) loss_classifier: 0.0079 (0.0101) loss_box_reg: 0.0029 (0.0039) loss_objectness: 0.0002 (0.0018) loss_rpn_box_reg: 0.0065 (0.0228) time: 0.1845 data: 0.0939 max mem: 1278\n",
"Epoch: [3] Total time: 0:00:10 (0.1871 s / it)\n",
"creating index...\n",
"index created!\n",
"Test: [ 0/16] eta: 0:00:01 model_time: 0.0312 (0.0312) evaluator_time: 0.0156 (0.0156) time: 0.0781 data: 0.0313 max mem: 1278\n",
"Test: [15/16] eta: 0:00:00 model_time: 0.0312 (0.0313) evaluator_time: 0.0000 (0.0020) time: 0.0626 data: 0.0283 max mem: 1278\n",
"Test: Total time: 0:00:01 (0.0636 s / it)\n",
"Averaged stats: model_time: 0.0312 (0.0313) evaluator_time: 0.0000 (0.0020)\n",
"Accumulating evaluation results...\n",
"DONE (t=0.02s).\n",
"IoU metric: bbox\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.898\n",
" Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 1.000\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
" Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.988\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.898\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.918\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.918\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.918\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.918\n",
"Epoch: [4] [ 0/58] eta: 0:00:12 lr: 0.005000 loss: 0.0219 (0.0219) loss_classifier: 0.0066 (0.0066) loss_box_reg: 0.0068 (0.0068) loss_objectness: 0.0003 (0.0003) loss_rpn_box_reg: 0.0081 (0.0081) time: 0.2187 data: 0.1250 max mem: 1278\n",
"Epoch: [4] [30/58] eta: 0:00:05 lr: 0.005000 loss: 0.0209 (0.0324) loss_classifier: 0.0109 (0.0103) loss_box_reg: 0.0040 (0.0039) loss_objectness: 0.0002 (0.0007) loss_rpn_box_reg: 0.0045 (0.0174) time: 0.1869 data: 0.0948 max mem: 1278\n",
"Epoch: [4] [57/58] eta: 0:00:00 lr: 0.005000 loss: 0.0215 (0.0336) loss_classifier: 0.0087 (0.0100) loss_box_reg: 0.0036 (0.0040) loss_objectness: 0.0004 (0.0014) loss_rpn_box_reg: 0.0040 (0.0183) time: 0.1851 data: 0.0946 max mem: 1278\n",
"Epoch: [4] Total time: 0:00:10 (0.1857 s / it)\n",
"creating index...\n",
"index created!\n",
"Test: [ 0/16] eta: 0:00:00 model_time: 0.0312 (0.0312) evaluator_time: 0.0000 (0.0000) time: 0.0625 data: 0.0312 max mem: 1278\n",
"Test: [15/16] eta: 0:00:00 model_time: 0.0312 (0.0352) evaluator_time: 0.0156 (0.0098) time: 0.0655 data: 0.0196 max mem: 1278\n",
"Test: Total time: 0:00:01 (0.0665 s / it)\n",
"Averaged stats: model_time: 0.0312 (0.0352) evaluator_time: 0.0156 (0.0098)\n",
"Accumulating evaluation results...\n",
"DONE (t=0.03s).\n",
"IoU metric: bbox\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.885\n",
" Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.885\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.903\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.903\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.903\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.903\n",
"Epoch: [5] [ 0/58] eta: 0:00:10 lr: 0.005000 loss: 0.0097 (0.0097) loss_classifier: 0.0052 (0.0052) loss_box_reg: 0.0021 (0.0021) loss_objectness: 0.0001 (0.0001) loss_rpn_box_reg: 0.0022 (0.0022) time: 0.1875 data: 0.0937 max mem: 1278\n",
"Epoch: [5] [30/58] eta: 0:00:05 lr: 0.005000 loss: 0.0194 (0.0291) loss_classifier: 0.0081 (0.0087) loss_box_reg: 0.0035 (0.0040) loss_objectness: 0.0002 (0.0015) loss_rpn_box_reg: 0.0031 (0.0149) time: 0.1885 data: 0.0989 max mem: 1278\n",
"Epoch: [5] [57/58] eta: 0:00:00 lr: 0.005000 loss: 0.0264 (0.0305) loss_classifier: 0.0085 (0.0093) loss_box_reg: 0.0026 (0.0038) loss_objectness: 0.0004 (0.0019) loss_rpn_box_reg: 0.0053 (0.0155) time: 0.1819 data: 0.0881 max mem: 1278\n",
"Epoch: [5] Total time: 0:00:10 (0.1863 s / it)\n",
"creating index...\n",
"index created!\n",
"Test: [ 0/16] eta: 0:00:00 model_time: 0.0469 (0.0469) evaluator_time: 0.0000 (0.0000) time: 0.0625 data: 0.0156 max mem: 1278\n",
"Test: [15/16] eta: 0:00:00 model_time: 0.0312 (0.0322) evaluator_time: 0.0000 (0.0000) time: 0.0626 data: 0.0303 max mem: 1278\n",
"Test: Total time: 0:00:01 (0.0626 s / it)\n",
"Averaged stats: model_time: 0.0312 (0.0322) evaluator_time: 0.0000 (0.0000)\n",
"Accumulating evaluation results...\n",
"DONE (t=0.03s).\n",
"IoU metric: bbox\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.900\n",
" Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.900\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.916\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.916\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.916\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.916\n",
"Epoch: [6] [ 0/58] eta: 0:00:12 lr: 0.005000 loss: 0.0224 (0.0224) loss_classifier: 0.0156 (0.0156) loss_box_reg: 0.0033 (0.0033) loss_objectness: 0.0001 (0.0001) loss_rpn_box_reg: 0.0034 (0.0034) time: 0.2077 data: 0.1140 max mem: 1278\n",
"Epoch: [6] [30/58] eta: 0:00:05 lr: 0.005000 loss: 0.0194 (0.0342) loss_classifier: 0.0100 (0.0095) loss_box_reg: 0.0038 (0.0033) loss_objectness: 0.0003 (0.0022) loss_rpn_box_reg: 0.0037 (0.0192) time: 0.1964 data: 0.1039 max mem: 1278\n",
"Epoch: [6] [57/58] eta: 0:00:00 lr: 0.005000 loss: 0.0241 (0.0341) loss_classifier: 0.0063 (0.0092) loss_box_reg: 0.0025 (0.0033) loss_objectness: 0.0002 (0.0022) loss_rpn_box_reg: 0.0042 (0.0194) time: 0.1841 data: 0.0926 max mem: 1278\n",
"Epoch: [6] Total time: 0:00:10 (0.1893 s / it)\n",
"creating index...\n",
"index created!\n",
"Test: [ 0/16] eta: 0:00:00 model_time: 0.0312 (0.0312) evaluator_time: 0.0000 (0.0000) time: 0.0625 data: 0.0313 max mem: 1278\n",
"Test: [15/16] eta: 0:00:00 model_time: 0.0312 (0.0313) evaluator_time: 0.0000 (0.0010) time: 0.0617 data: 0.0294 max mem: 1278\n",
"Test: Total time: 0:00:00 (0.0617 s / it)\n",
"Averaged stats: model_time: 0.0312 (0.0313) evaluator_time: 0.0000 (0.0010)\n",
"Accumulating evaluation results...\n",
"DONE (t=0.02s).\n",
"IoU metric: bbox\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.911\n",
" Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.988\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.911\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.927\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.927\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.927\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.927\n",
"Epoch: [7] [ 0/58] eta: 0:00:11 lr: 0.000500 loss: 0.0124 (0.0124) loss_classifier: 0.0071 (0.0071) loss_box_reg: 0.0025 (0.0025) loss_objectness: 0.0001 (0.0001) loss_rpn_box_reg: 0.0027 (0.0027) time: 0.2031 data: 0.1094 max mem: 1278\n",
"Epoch: [7] [30/58] eta: 0:00:05 lr: 0.000500 loss: 0.0203 (0.0266) loss_classifier: 0.0089 (0.0091) loss_box_reg: 0.0027 (0.0033) loss_objectness: 0.0002 (0.0010) loss_rpn_box_reg: 0.0031 (0.0132) time: 0.1930 data: 0.0999 max mem: 1278\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"Epoch: [7] [57/58] eta: 0:00:00 lr: 0.000500 loss: 0.0234 (0.0282) loss_classifier: 0.0050 (0.0084) loss_box_reg: 0.0018 (0.0027) loss_objectness: 0.0003 (0.0016) loss_rpn_box_reg: 0.0031 (0.0154) time: 0.1847 data: 0.0915 max mem: 1278\n",
"Epoch: [7] Total time: 0:00:10 (0.1896 s / it)\n",
"creating index...\n",
"index created!\n",
"Test: [ 0/16] eta: 0:00:00 model_time: 0.0312 (0.0312) evaluator_time: 0.0000 (0.0000) time: 0.0625 data: 0.0312 max mem: 1278\n",
"Test: [15/16] eta: 0:00:00 model_time: 0.0312 (0.0325) evaluator_time: 0.0000 (0.0049) time: 0.0638 data: 0.0264 max mem: 1278\n",
"Test: Total time: 0:00:01 (0.0638 s / it)\n",
"Averaged stats: model_time: 0.0312 (0.0325) evaluator_time: 0.0000 (0.0049)\n",
"Accumulating evaluation results...\n",
"DONE (t=0.02s).\n",
"IoU metric: bbox\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.908\n",
" Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.988\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.908\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.925\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.925\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.925\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.925\n",
"Epoch: [8] [ 0/58] eta: 0:00:10 lr: 0.000500 loss: 0.0187 (0.0187) loss_classifier: 0.0130 (0.0130) loss_box_reg: 0.0040 (0.0040) loss_objectness: 0.0001 (0.0001) loss_rpn_box_reg: 0.0016 (0.0016) time: 0.1886 data: 0.1105 max mem: 1278\n",
"Epoch: [8] [30/58] eta: 0:00:05 lr: 0.000500 loss: 0.0151 (0.0214) loss_classifier: 0.0062 (0.0073) loss_box_reg: 0.0018 (0.0021) loss_objectness: 0.0002 (0.0017) loss_rpn_box_reg: 0.0025 (0.0103) time: 0.1897 data: 0.0960 max mem: 1278\n",
"Epoch: [8] [57/58] eta: 0:00:00 lr: 0.000500 loss: 0.0168 (0.0231) loss_classifier: 0.0090 (0.0079) loss_box_reg: 0.0026 (0.0025) loss_objectness: 0.0002 (0.0016) loss_rpn_box_reg: 0.0028 (0.0111) time: 0.1951 data: 0.1011 max mem: 1278\n",
"Epoch: [8] Total time: 0:00:11 (0.1910 s / it)\n",
"creating index...\n",
"index created!\n",
"Test: [ 0/16] eta: 0:00:01 model_time: 0.0313 (0.0313) evaluator_time: 0.0000 (0.0000) time: 0.0625 data: 0.0313 max mem: 1278\n",
"Test: [15/16] eta: 0:00:00 model_time: 0.0312 (0.0312) evaluator_time: 0.0000 (0.0010) time: 0.0616 data: 0.0284 max mem: 1278\n",
"Test: Total time: 0:00:00 (0.0616 s / it)\n",
"Averaged stats: model_time: 0.0312 (0.0312) evaluator_time: 0.0000 (0.0010)\n",
"Accumulating evaluation results...\n",
"DONE (t=0.02s).\n",
"IoU metric: bbox\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.908\n",
" Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.985\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.908\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.928\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.928\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.928\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.928\n",
"Epoch: [9] [ 0/58] eta: 0:00:13 lr: 0.000500 loss: 0.0217 (0.0217) loss_classifier: 0.0152 (0.0152) loss_box_reg: 0.0044 (0.0044) loss_objectness: 0.0001 (0.0001) loss_rpn_box_reg: 0.0020 (0.0020) time: 0.2318 data: 0.1406 max mem: 1278\n",
"Epoch: [9] [30/58] eta: 0:00:05 lr: 0.000500 loss: 0.0204 (0.0228) loss_classifier: 0.0058 (0.0078) loss_box_reg: 0.0017 (0.0022) loss_objectness: 0.0002 (0.0014) loss_rpn_box_reg: 0.0025 (0.0115) time: 0.1823 data: 0.0902 max mem: 1278\n",
"Epoch: [9] [57/58] eta: 0:00:00 lr: 0.000500 loss: 0.0159 (0.0212) loss_classifier: 0.0076 (0.0081) loss_box_reg: 0.0019 (0.0024) loss_objectness: 0.0001 (0.0013) loss_rpn_box_reg: 0.0027 (0.0094) time: 0.1857 data: 0.0947 max mem: 1278\n",
"Epoch: [9] Total time: 0:00:10 (0.1882 s / it)\n",
"creating index...\n",
"index created!\n",
"Test: [ 0/16] eta: 0:00:01 model_time: 0.0313 (0.0313) evaluator_time: 0.0156 (0.0156) time: 0.0636 data: 0.0167 max mem: 1278\n",
"Test: [15/16] eta: 0:00:00 model_time: 0.0312 (0.0322) evaluator_time: 0.0000 (0.0029) time: 0.0616 data: 0.0245 max mem: 1278\n",
"Test: Total time: 0:00:00 (0.0616 s / it)\n",
"Averaged stats: model_time: 0.0312 (0.0322) evaluator_time: 0.0000 (0.0029)\n",
"Accumulating evaluation results...\n",
"DONE (t=0.03s).\n",
"IoU metric: bbox\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.904\n",
" Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 1.000\n",
" Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.985\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.904\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.923\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.923\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.923\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = -1.000\n",
" Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.923\n"
]
}
],
"outputs": [],
"source": [
"# Fine-tune model. After each epoch prints the accuracy on the validation set.\n",
"detector.fit(EPOCHS, lr=LEARNING_RATE, print_freq=30)"