-
从零开始实现的Tensorflow 2 Faster-RCNN支持采用MobileNetV2和VGG16作为主干网络进行批处理
资源介绍
更快的RCNN
这是从零开始支持到批处理的tensorflow Faster-RCNN实现。 为了易于理解,尝试以最简单的方式创建所有方法。 在实施过程中执行的大多数操作是按照和信息库中所述进行的。
它是使用tensorflow 2.0实现和测试的。
支持和主干。
用法
使用在虚拟环境中创建的项目模型。 您还可以使用创建所需的虚拟环境。
要创建虚拟环境(tensorflow-2 gpu环境):
conda env create -f environment.yml
有两个不同的主干网,第一个是传统vgg16主干网,第二个和默认主干网是mobilenet_v2 。 您可以轻松地与--backbone参数一起指定要使用的主干。 默认主干是mobilenet_v2 。
训练和测试Faster-RCNN模型:
python faster_rcnn_trainer.py --back