search

facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
https://detectron2.readthedocs.io/en/latest/
Apache License 2.0
30.46k stars 7.48k forks source link

How to run inference with test-time augmentation on Mask R-CNN X152 model? #293

Closed BaiGungdong closed 4 years ago

BaiGungdong commented 4 years ago
[11/11 07:40:58] d2.evaluation.evaluator INFO: Start inference on 323 images [11/11 07:40:58] d2.evaluation.evaluator INFO: aaaaaaaaaaaa323 [11/11 07:42:34] d2.evaluation.evaluator INFO: Total inference time: 0:01:29 (0.279874 s / img per device, on 4 devices) [11/11 07:42:34] d2.evaluation.evaluator INFO: Total inference pure compute time: 0:01:27 (0.276265 s / img per device, on 4 devices) [11/11 07:42:35] d2.evaluation.coco_evaluation INFO: Preparing results for COCO format ... [11/11 07:42:35] d2.evaluation.coco_evaluation INFO: Saving results to ./output_cascade_X_153_FPN_siz2/inference/coco_instances_results.json [11/11 07:42:35] d2.evaluation.coco_evaluation INFO: Evaluating predictions ... [11/11 07:42:36] d2.evaluation.coco_evaluation INFO: Evaluation results for bbox: AP AP50 AP75 APs APm APl
84.673 93.418 90.251 87.581 88.611 78.224

[11/11 07:42:36] d2.engine.defaults INFO: Evaluation results for my_dataset_val2 in csv format: [11/11 07:42:36] d2.evaluation.testing INFO: copypaste: Task: bbox [11/11 07:42:36] d2.evaluation.testing INFO: copypaste: AP,AP50,AP75,APs,APm,APl [11/11 07:42:36] d2.evaluation.testing INFO: copypaste: 84.6727,93.4183,90.2510,87.5808,88.6114,78.2240 [11/11 07:42:36] d2.trainer INFO: Running inference with test-time augmentation ... [11/11 07:42:36] d2.data.datasets.coco INFO: Loaded 1292 images in COCO format from ./datasets/gongjing2/val.json [11/11 07:42:36] d2.evaluation.evaluator INFO: Start inference on 323 images [11/11 07:42:36] d2.evaluation.evaluator INFO: aaaaaaaaaaaa323 [11/11 07:46:48] d2.evaluation.evaluator INFO: Total inference time: 0:04:02 (0.761006 s / img per device, on 4 devices) [11/11 07:46:48] d2.evaluation.evaluator INFO: Total inference pure compute time: 0:04:01 (0.757913 s / img per device, on 4 devices) [11/11 07:46:54] d2.evaluation.coco_evaluation INFO: Preparing results for COCO format ... [11/11 07:46:54] d2.evaluation.coco_evaluation INFO: Saving results to ./output_cascade_X_153_FPN_siz2/inference_TTA/coco_instances_results.json [11/11 07:46:54] d2.evaluation.coco_evaluation INFO: Evaluating predictions ... [11/11 07:46:54] d2.evaluation.coco_evaluation WARNING: No predictions from the model! Set scores to -1 [11/11 07:46:54] d2.engine.defaults INFO: Evaluation results for my_dataset_val2 in csv format: [11/11 07:46:54] d2.evaluation.testing INFO: copypaste: Task: bbox [11/11 07:46:54] d2.evaluation.testing INFO: copypaste: AP,AP50,AP75,APs,APm,APl [11/11 07:46:54] d2.evaluation.testing INFO: copypaste: -1.0000,-1.0000,-1.0000,-1.0000,-1.0000,-1.0000 [11/11 07:55:05] detectron2 INFO: Rank of current process: 0. World size: 4

ppwwyyxx commented 4 years ago 
python tools/train_net.py --config-file configs/Misc/cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml --num-gpus 8 --eval-only TEST.AUG.ENABLED True MODEL.WEIGHTS https://dl.fbaipublicfiles.com/detectron2/Misc/cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv/18131413/model_0039999_e76410.pkl
BaiGungdong commented 4 years ago

I had set TEST.AUG.ENABLED to true，but it was not working

ppwwyyxx commented 4 years ago

Please include details about the problem following the issue template.