使用 fn_output_signature 代替 stop 问题

Question

我的输入；

python model_main_tf2.py --model_dir=models\my_ssd_mobilenet_v2_fpnlite --pipeline_config_path=models\my_ssd_mobilenet_v2_fpnlite\pipeline.config

输出；

prnt.sc/104ecjv

我的学习在这个阶段停止了，一动不动。

我的参考； https://github.com/armaanpriyadarshan/Training-a-Custom-TensorFlow-2.x-Object-Detector请帮帮我。

Answer 1

有同样的问题，并在此浪费了很多时间。这是我的解决方法

1. 确保您的 csv 中没有任何宽度、高度的负值
2. 对生成的 csv 文件进行一些完整性检查。
3. 如果脚本中没有正确检查，则 tf_records.records 没有信息 (0kb)
4. 在我的一个案例中，我正在使用 pip 安装软件包，并且正在重置在 google colab 中分配的 gpu。

也可以参考下面的链接。

https://github.com/tensorflow/models/issues/9106

https://bleepcoder.com/models/627797941/tensorflow-python-framework-errors-impl-invalidargumenterror_8595