Faster Rcnn 代码解读之 roi_data_layer/layer.py

最新推荐文章于 2020-03-18 21:16:57 发布

多一些不为什么的坚持

最新推荐文章于 2020-03-18 21:16:57 发布

阅读量1.3k

点赞数

CC 4.0 BY-SA版权

分类专栏： FasterRcnn Faster Rcnn

本文链接：https://blog.youkuaiyun.com/qq_33193309/article/details/98877244

本文深入剖析Faster R-CNN目标检测算法中roi_data_layer/layer.py模块的实现细节，揭示其在训练过程中的关键功能，包括ROI池化、数据预处理等步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# --------------------------------------------------------
# Fast R-CNN
# Copyright (c) 2015 Microsoft
# Licensed under The MIT License [see LICENSE for details]
# Written by Ross Girshick and Xinlei Chen
# --------------------------------------------------------

"""The data layer used during training to train a Fast R-CNN network.

RoIDataLayer implements a Caffe Python layer.
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from model.config import cfg
from roi_data_layer.minibatch import get_minibatch
import numpy as np
import time


class RoIDataLayer(object):
    """Fast R-CNN data layer used for training."""

    def __init__(self, roidb, num_classes, random=False):
        """Set the roidb to be used by this layer during training."""
        self._roidb = roidb
        self._num_classes = num_classes
        # Also set a random flag
        self._random = random
        self._shuffle_roidb_i