我正在尝试训练Mask R-CNN进行实例细分。哪里有一些可用的预训练模型。这些权重是用于整个神经网络还是仅用于编码器/骨干(例如resnet50)?还有使用imagenet或coco进行初始化的权重。对于后者,解码器中的所有权重都是随机的吗?
答案 0 :(得分:0)
以下是一些最好的Mask RCNN实现,
1. https://github.com/matterport/Mask_RCNN
2. https://github.com/CharlesShang/FastMaskRCNN
3. https://github.com/multimodallearning/pytorch-mask-rcnn
4. https://github.com/wannabeOG/Mask-RCNN
预训练权重和初始化权重通常适用于整个网络,而不适用于骨干网。
建议您先进行研究,然后再发布stackoverflow。请转到github并搜索Mask RCNN,您将找到大量的仓库。