Skip to content
This repository has been archived by the owner on May 5, 2020. It is now read-only.

Latest commit

 

History

History

mask_rcnn_2go

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 

MaskRCNN2GO Model

Owner: Peizhao Zhang (stzpz@fb.com)

Model

  • MaskRCNN2GO (bbox + segmentation) with 81 classes
  • float32 and int8 models
  • Trained on COCO 2014 dataset
  • int8 model fine-tuned with fake-quantization

Performance

  • Evaluation dataset: COCO 2014 minival
  • Metric: mAP[IoU=0.50:0.95] and latency (in microseconds)
  • Proposals: 3000/100 (pre/post nms)
  • The backend for evaluation is QNNPACK
  • The unit of mensured latency is microseconds
  • Latency is evaluated on Samsung S8 (SM-G950U-7.0-24)
Model Bbox Segmentation Latency Median Latency MAD
float32 25.1 21.6 - -
int8 24.8 21.6 150088 78744.55

Input

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

Evaluation

  • Download COCO 2014 minival dataset
  • OpenCV must be installed for image preprocessing
  • Run run.sh [coco_dir] [model]
  • The model argument can be either fp32 for float 32 or int8

Model source

  • f93520960
  • f96110081:934610037

Acknowledgement

Thanks a lot for the help from Carole-Jean Wu, Fei Sun, Yiming Wu, Yanghan Wang and Zhizhen Qin.