摘要：将一个tensor分到多个GPU上：torch.cuda.comm.scatter 阅读全文

摘要：faster-rcnn的整体流程比较复杂，尤其是数据的预处理部分，流程比较繁琐。我写faster-rcnn系列文章的目的是对该算法的原始版本有个整体的把握，如果需要使用该算法做一些具体的任务，推荐使用mmdetection框架，该框架使用PyTorch写成，相比于原始的基于caffe python接 阅读全文

摘要：这一节讲述roi-data层，和这一层有关的结构图如下： roi-data层的prototxt定义如下： 这一层接着上一节的proposal层，主要的操作是对RPN阶段输出的proposal做进一步的筛选，从中选出背景和前景，得到rois，然后计算rois的回归目标。因为第二阶段的rois回归是和类 阅读全文

摘要：这一节讲述proposal层，和这一层有关的结构图如下： proposal层的prototxt定义如下： 这一层的功能是对卷积网络中RPN输出的bbox_deltas, scores做后处理，主要步骤如下： 1、同上节中的第一步，生成anchor； 2、将anchor和RPN网络输出的bbox_de 阅读全文

摘要：这一节讲述rpn-data层，和这一层有关的结构图如下： rpn-data层的prototxt定义如下： 这一层的主要工作如下： 一、生成anchor，并将超出图像区域的anchor去除，得到有效的anchor； 二、给每一个anchor分配label，-1表示忽略该anchor，0表示背景，1表示 阅读全文