我的输入;
python model_main_tf2.py --model_dir=modelsmy_ssd_mobilenet_v2_fpnlite --pipeline_config_path=modelsmy_ssd_mobilenet_v2_fpnlitepipeline.config
输出;
prnt.sc/104 ecjv
我的学习就停在这个阶段,不动了。
我的参考;https://github.com/armaanpriyadarshan/Training-a-Custom-TensorFlow-2.x-Object-Detector请帮帮我。
有同样的问题,浪费了很多时间。下面是我解决它的方法
- 确保你的csv 中宽度和高度没有负值
- 对生成的csv文件做一些完整性检查
- 如果脚本中没有适当的检查,则tf_records。记录没有信息(0kb)
- 在我的一个案例中,我正在使用pip安装软件包,并且正在重置google colab中分配的gpu。
请参考下面的链接。
https://github.com/tensorflow/models/issues/9106
https://bleepcoder.com/models/627797941/tensorflow-python-framework-errors-impl-invalidargumenterror_8595