未加星标

Architecture of Mask RCNN A PyTorch Implementation

字体大小 | |
[开发(python) 所属分类 开发(python) | 发布者 店小二05 | 时间 2018 | 作者 红领巾 ] 0人收藏点击收藏
Mask-RCNN

A PyTorch implementation of the architecture of Mask RCNN, serves as an introduction to working with PyTorch

Decription of folders model.py includes the models of ResNet and FPN which were already implemented by the authors of the papers and reproduced in this implementation nms and RoiAlign are taken from Robb Girshick's implementation of faster RCNN Focal loss has been added to this implementtaion on lieu of better results as evidenced by the paper on RetinaNets Mask-RCNN model:
Architecture of Mask RCNN   A PyTorch Implementation
Features: The part of the network responsible for bounding box detection derives it's inspiration from the faster RCNN model having a RPN working in tandem with a ConvNet The pooling layers present in the ConvNet round down or round up to the nearest integer when the stride is not a divisor of the receptive field, which tends to either lose or assume "information" from the image respectively at the non integral points. ROI align was proposed to deal with this, wherein bilinear interpolation is used to detect the values at the non integral values of the pixels Using a more complex interpolation scheme( cubic interpolation -> 16 additional features) offers a slightly better result when this model was tested, however not enough to justify the additional complexity Cross entropy loss when summed over a huge number of proposals tends to take a huge value for proposals that have a high confidence metric thereby dwarfing the contribution from the proposals of interest. Focal Loss was proposed to do away with this problem However Focal loss gives much better results with single stage networks. This is because a two stage network has some discriminative policy to deal with this class imbalance something which the single stage networks don't enjoy.

If you find any issue in this repsoritory, feel free to fork this repository and submit a PR with the necessary changes

本文开发(python)相关术语:python基础教程 python多线程 web开发工程师 软件开发工程师 软件开发流程

tags: RCNN,Mask,model,implementation,Focal
分页:12
转载请注明
本文标题:Architecture of Mask RCNN A PyTorch Implementation
本站链接:https://www.codesec.net/view/586934.html


1.凡CodeSecTeam转载的文章,均出自其它媒体或其他官网介绍,目的在于传递更多的信息,并不代表本站赞同其观点和其真实性负责;
2.转载的文章仅代表原创作者观点,与本站无关。其原创性以及文中陈述文字和内容未经本站证实,本站对该文以及其中全部或者部分内容、文字的真实性、完整性、及时性,不作出任何保证或承若;
3.如本站转载稿涉及版权等问题,请作者及时联系本站,我们会及时处理。
登录后可拥有收藏文章、关注作者等权限...
技术大类 技术大类 | 开发(python) | 评论(0) | 阅读(34)