Find file History
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
code Add MaskRCNN2Go model to Caffe2 Model zoo Dec 13, 2018
model Add MaskRCNN2Go model to Caffe2 Model zoo Dec 13, 2018
README.md Add MaskRCNN2Go model to Caffe2 Model zoo Dec 13, 2018
run_eval.sh Remove run output and update the script Dec 13, 2018

README.md

MaskRCNN2GO MLPerf submission

Owner: Peizhao Zhang (stzpz@fb.com)

Model: MaskRCNN2GO (bbox + segmentation) float32 and int8

Datasets used for evaluation: COCO 2014 minival

Input:

  • data (1, 3, H, W), min(H, W) = 320, BGR in range [0, 255]
  • im_info (1, 3) [scaled_height, scaled_width, scale]

Proposals (pre nms, post nms): 3000/100

Accuracy(mAP[IoU=0.50:0.95]): float32: 24.5 (bbox), 21.7 (segmentation) int8: 23.5 (bbox), 20.9 (segmentation)

Evaluation:

  • Download COCO 2014 minival dataset
  • Update path in run_eval.sh
  • Run run_eval.sh

Acknowledgement

Thanks a lot for the help from Carole-Jean Wu and Fei Sun.