18150167970 / knowledge-distillation Goto Github PK

After incremental learning with your code and default parameters, the precision of all the classes is low.
======user config======== {'VOC_BBOX_LABEL_NAMES': ('aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor'), 'VOC_BBOX_LABEL_NAMES_all': ('aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor'), 'caffe_pretrain': True, 'caffe_pretrain_path': 'pretrained_model/faster_rcnn_1_8_9873_remove.pth', 'data': 'voc', 'datatxt': 'trainval', 'debug_file': 'tmp/debugf', 'env': 'faster-rcnn', 'epoch': 100, 'index': 19, 'is_distillation': True, 'load_path': 'pretrained_model/faster_rcnn_1_8_9873_teacher19.pth', 'lr': 1e-05, 'lr_decay': 0.1, 'max_size': 1000, 'min_size': 600, 'num_workers': 8, 'only_use_cls_distillation': False, 'plot_every': 40, 'port': 8097, 'predict_socre': 0.05, 'pretrained_model': 'vgg16', 'roi_sigma': 1.0, 'rpn_sigma': 3.0, 'test_num': 5000, 'test_num_workers': 8, 'test_path': './pretrained_model/faster_rcnn_1_8_9873_remove.pth', 'testtxt': 'test', 'threshold': 0.5, 'use_adam': False, 'use_chainer': False, 'use_drop': False, 'use_hint': True, 'voc_data_dir': 'data/VOCdevkit2007_19+1/VOC2007_tm', 'voc_test_data_dir': 'data/VOCdevkit2007/VOC2007', 'weight_decay': 5e-05} ==========end============ load data =======================VOCBboxDataset========================== tvmonitor =======================VOCBboxDataset========================== tvmonitor =======================VOCBboxDataset_test_all========================== ('aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor') Setting up a new session... model construct completed load own model load test model from ./checkpoints/fasterrcnn_11161633_0.53489727386923.pth 4952it [36:57, 2.23it/s] lr:1e-05,ap:[0.00124533 0.00332059 0. 0.0013485 0.03199189 0.00131106 0.08138579 0.09791409 0.1394366 0.00186786 0.05367795 0.14745307 0.00500412 0.00044636 0.20192489 0.06652 0.00124802 0.14069406 0.00078145 0.1327586 ], map:0.05551651161754526,loss:{'rpn_loc_loss': nan, 'rpn_cls_loss': nan, 'roi_loc_loss': nan, 'roi_cls_loss': nan, 'hint_loss': nan, 'scores_loss': nan, 'total_loss': nan}