search

facebookresearch / maskrcnn-benchmark

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
MIT License
9.29k stars 2.5k forks source link

what is the difference between make_roi_box_feature_extractor and make_roi_mask_feature_extractor? #420

Open JoyHuYY1412 opened 5 years ago

JoyHuYY1412 commented 5 years ago

❓ Questions and Help

I think they all choose a selected layer and use ROIAlign to get the feature. So why do you write two files?

fmassa commented 5 years ago

It's just to have some clear separation between boxes and masks. Functionally, they are indeed similar, but the return arguments are expected to have different data (in the mask branch, a mask field is added to the BoxList).