Faster-Rcnn demo.py解析

最新推荐文章于 2024-06-19 09:41:12 发布

航子_harder

最新推荐文章于 2024-06-19 09:41:12 发布

阅读量7k

点赞数 3

分类专栏：深度学习文章标签：源代码 python faster-rcn

本文链接：https://blog.csdn.net/u013129427/article/details/74652893

版权

本文详细解析 Faster-RCNN 的 Python 实现，包括 `vis_detections` 和 `demo` 模块，用于画出测试图片的边界框。同时介绍了如何使用 argparse 模块解析命令行参数，以及 IoU 和非极大值抑制等关键概念。

摘要由CSDN通过智能技术生成

import _init_paths
from fast_rcnn.config import cfg
#im_detect 生成RPN候选框
from fast_rcnn.test import im_detect
#nms 进行非极大值抑制
from fast_rcnn.nms_wrapper import nms
from utils.timer import Timer
import matplotlib.pyplot as plt
import numpy as np
import scipy.io as sio
import caffe, os, sys, cv2
#引入argparse， 它python用于解析命令行参数和选项的
#标准模块，用于解析命令行参数
import argparse

CLASSES = ('__background__',
           'aeroplane', 'bicycle', 'bird', 'boat',
           'bottle', 'bus', 'car', 'cat', 'chair',
           'cow', 'diningtable', 'dog', 'horse',
           'motorbike', 'person', 'pottedplant',
           'sheep', 'sofa', 'train', 'tvmonitor')


NETS = {
  'vgg16': ('VGG16',
                  'VGG16_faster_rcnn_final.caffemodel'),
        'zf': ('ZF',
                  'ZF_faster_rcnn_final.caffemodel')}