ObjectDetectionTraining:用于TensorFlow对象检测模型的训练和重新训练脚本 源码

模型训练 用于在COCO类的选择性子集上(重新)训练Tensorflow Zoo模型的脚本 依存关系 运行./install_deps.sh以安装必要的依赖项。 如果要在非Jetson设备上安装,请设置环境变量NOT_NVIDIA=true 。 该脚本创建了一个Python虚拟环境,您可以通过运行source env/bin/activate 。 生成输入数据 Tensorflow API使用一种特殊类型的文件作为其输入图像: 格式。 在可以找到将COCO数据集转换为正确的TFRecord文件格式的完整文档。 新数据集 虽然COCO数据集非常庞大,并且包含我们关注的有限子集(人,汽车,卡车等)中每个类的许多示例,但该数据集并没有包含许多来自不同角度的奇怪图像,例如无人机或监控镜头捕获的物体的俯视透视图。 以下是一些有希望的数据集,它们可以帮助缓解此问题: 如果证明这些(或其他类似)数据