1. /tools/train_net.py ( Faster-RCNN_TF代码解读)

最新推荐文章于 2022-02-25 09:16:49 发布

风轻扬逍遥子

最新推荐文章于 2022-02-25 09:16:49 发布

阅读量1.3k

点赞数

分类专栏： faster-r-cnn 文章标签： faster r-cnn tensorflow 代码解读

本文链接：https://blog.csdn.net/u014256231/article/details/79696108

版权

本文主要解读Faster R-CNN_TF的train_net.py代码，包括从imdb、roidb到network的构建及训练过程。通过get_imdb、get_training_roidb、get_network和train_net等关键函数，详细解析了模型的训练流程。

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

1. /tools/train_net.py

我用的Faster-RCNN是tensorflow版本，github地址：Faster-RCNN_TF

代码运行是从train_net.py进入的。

调用函数链接：

imdb = get_imdb(args.imdb_name)中的get_imdb函数在/lib/datasets/factor.py中。
roidb = get_training_roidb(imdb)中的get_training_roidb函数在train.py中。
network = get_network(xxxxx)中的get_network函数在/lib/networks/factory.py中。
train_net(xxxxx)中的train_net在函数/lib/fast_rcnn/train.py中。

代码解读：

#!/usr/bin/env python

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

"""Train a Fast R-CNN network on a region of interest database."""

import _init_paths
from fast_rcnn.train import get_training_roidb, train_net
from fast_rcnn.config import cfg,cfg_from_file, cfg_from_list, get_output_dir
from datasets.factory import get_imdb