Tensorflow/models 中的 object_detection 模型训练时遇到的问题

谷歌维护的与Tensorflow配套的models源码库包含了大量前沿算法实践。

在调用 object_detection/model_main.py时可能会遇到以下错误:

tensorflow.python.framework.errors_impl.NotFoundError: Unsuccessful TensorSliceReader constructor: Failed to find any matching files for ???/model.ckpt

model.ckpt这个文件似乎只在.config文件中见到:

如在object_detection/samples/configs/ssd_mobilenet_v2_coco.config中有:

fine_tune_checkpoint: "PATH_TO_BE_CONFIGURED/model.ckpt"
fine_tune_checkpoint_type: "detection"

fine_tune_checkpoint:此处应该为空白,否则会导致无法训练的情况。(参考这里)
即:

fine_tune_checkpoint: ""
fine_tune_checkpoint_type: "detection"

.ckpt文件是训练时生成的。当进行模型的初步训练时,该文件还不存在。推测该项应当供模型微调时使用。

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容