Faster Rcnn 代码解读之 test.py

最新推荐文章于 2021-03-16 21:38:58 发布

原创

最新推荐文章于 2021-03-16 21:38:58 发布 · 1.3k 阅读

5 ·

CC 4.0 BY-SA版权

本文详细解析Faster R-CNN目标检测框架中的test.py文件，深入理解其实现过程和关键算法步骤，帮助读者掌握目标检测技术。

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

# --------------------------------------------------------
# Tensorflow Faster R-CNN
# Licensed under The MIT License [see LICENSE for details]
# Written by Xinlei Chen
# --------------------------------------------------------
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import cv2
import numpy as np

try:
    import cPickle as pickle
except ImportError:
    import pickle
import os
import math

from utils.timer import Timer
from utils.blob import im_list_to_blob

from model.config import cfg, get_output_dir
from model.bbox_transform import clip_boxes, bbox_transform_inv
from model.nms_wrapper import nms

'''用于测试的时候调用模型进行测试,在tools/test_net.py调用'''


def _get_image_blob(im):
    """Converts an image into a network input.
    将图片转换成网络的输入格式
    Arguments:
      im (ndarray): a color image in BGR order
    Returns:
      blob (ndarray): a data blob holding an image pyramid
      im_scale_factors (list): list of image scales (relative to im) used
        in the image pyramid
    """

    im_orig = im.astype(np.float32, copy=True)
    # 去均值
    im_orig -= cfg.PIXEL_MEANS
    # 提取最大、最小边
    im_shape = im_orig.shape
    im_size_min = np.min(im_shape[0:2])
    im_size_max = np.max(im_shape[0:2])

    processed_ims = []
    im_scale_factors = []

    for target_size in cfg.TEST.SCALES:
        # 缩放比例
        im_scale = float(target_size) / float(im_size_min)
        # Prevent the biggest axis from being more than MAX_SIZE
        if np.round(im_scale *