一、模型介绍
MaskRCNN模型主要包括四个部分,backbone、RPN、ROIHeads(Box roi head、Mask roi heads)、PostProcessing(推理时会用到)。网络结构上Mask Rcnn部分初始化了Mask roi heads,
#...\Miniconda3\Lib\site-packages\torchvision\models\detection\mask_rcnn.py
if mask_roi_pool is None:
MaskRCNN模型主要包括四个部分,backbone、RPN、ROIHeads(Box roi head、Mask roi heads)、PostProcessing(推理时会用到)。网络结构上Mask Rcnn部分初始化了Mask roi heads,
#...\Miniconda3\Lib\site-packages\torchvision\models\detection\mask_rcnn.py
if mask_roi_pool is None: