Duankaiwen / CenterNet

Codes for our paper "CenterNet: Keypoint Triplets for Object Detection" .
MIT License
1.86k stars 384 forks source link

How to get map when I train with voc2007 #124

Open Glutton-zh opened 4 years ago

Glutton-zh commented 4 years ago

i train on voc2007 and when i test it

$ python3 test.py CenterNet-52 --testiter 225000 --split testing cfg_file: config/CenterNet-52.json loading all datasets... split: testdev loading from cache file: cache/coco_testdev2017.pkl loading annotations into memory... Done (t=0.08s) creating index... index created! system config... {'batch_size': 2, 'cache_dir': 'cache', 'chunk_sizes': [2], 'config_dir': 'config', 'data_dir': './data', 'data_rng': <mtrand.RandomState object at 0x7fe66ca1a8b8>, 'dataset': 'MSCOCO', 'decay_rate': 10, 'display': 5, 'learning_rate': 0.00025, 'max_iter': 225000, 'nnet_rng': <mtrand.RandomState object at 0x7fe66ca1a900>, 'opt_algo': 'adam', 'prefetch_size': 6, 'pretrain': None, 'result_dir': 'results', 'sampling_function': 'kp_detection', 'snapshot': 5000, 'snapshot_name': 'CenterNet-52', 'stepsize': 150000, 'test_split': 'testdev', 'train_split': 'trainval', 'val_iter': 500, 'val_split': 'minival', 'weight_decay': False, 'weight_decay_rate': 1e-05, 'weight_decay_type': 'l2'} db config... {'ae_threshold': 0.5, 'border': 128, 'categories': 20, 'data_aug': True, 'gaussian_bump': True, 'gaussian_iou': 0.7, 'gaussian_radius': -1, 'input_size': [511, 511], 'kp_categories': 1, 'lighting': True, 'max_per_image': 100, 'merge_bbox': False, 'nms_algorithm': 'exp_soft_nms', 'nms_kernel': 3, 'nms_threshold': 0.5, 'output_sizes': [[128, 128]], 'rand_color': True, 'rand_crop': True, 'rand_pushes': False, 'rand_samples': False, 'rand_scale_max': 1.4, 'rand_scale_min': 0.6, 'rand_scale_step': 0.1, 'rand_scales': array([0.6, 0.7, 0.8, 0.9, 1. , 1.1, 1.2, 1.3]), 'special_crop': False, 'test_scales': [1], 'top_k': 70, 'weight_exp': 8} loading parameters at iteration: 225000 building neural network... module_file: models.CenterNet-52 total parameters: 104797892 loading parameters... loading model from cache/nnet/CenterNet-52/CenterNet-52_225000.pkl locating kps: 0%| | 0/4952 [00:00<?, ?it/s]/home/zhanghan/anaconda3/envs/CornerNet_Lite/lib/python3.7/site-packages/torch/nn/modules/upsampling.py:129: UserWarning: nn.Upsample is deprecated. Use nn.functional.interpolate instead. warnings.warn("nn.{} is deprecated. Use nn.functional.interpolate instead.".format(self.name)) locating kps: 100%|█████████████████████████| 4952/4952 [13:40<00:00, 6.04it/s]

that's all. i just get a result.json in my /test where is the function to calculate the map that i can get my Map

adrian-wl commented 4 years ago

I met the same puzzle