Tensorflow-Faster代码解析: test.py

test.py 文件位于 libmodel 目录下,主要包含 test_net.py 函数,该函数在测试 Faster 模型时被 tools/test_net.py 调用。此文件负责调用模型并保存测试结果,其内部逻辑清晰,每一步操作都有详细注释。

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

 

lib\model\test.py里主要是test_net.py,用于faster测试的时候调用模型进行测试,并对结果进行保存。在tools/test_net.py里被调用。函数相对简单,但确实是测试时的整个流程了。基本每一句都写得很清楚了。

# --------------------------------------------------------
# Tensorflow Faster R-CNN
# Licensed under The MIT License [see LICENSE for details]
# Written by Xinlei Chen
#解析:nansbas
# --------------------------------------------------------
#-*- coding: utf-8 -*-
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

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
  """
  #图像转换为float32。注意是bgr顺序,不是rgb顺序
  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 = []
  #cfg.TEST.SCALES=600
  for target_size in cfg.T
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值