登录 注册
当前位置:主页 > 资源下载 > 27 > 从零开始实现的Tensorflow 2 Faster-RCNN支持采用MobileNetV2和VGG16作为主干网络进行批处理

从零开始实现的Tensorflow 2 Faster-RCNN支持采用MobileNetV2和VGG16作为主干网络进行批处理

  • 更新:2024-12-06 10:33:20
  • 大小:27KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

更快的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