跑TDD-net遇到的一些坑

跑TDD-net遇到的一些坑,第1张

1、tensorflow不要用requirement里的2.5.1,一定要用1。因为不兼容,尝试过debug一下午都没法。

2、cudnn、gcc、tensorflow、cuda这四者严格按照相关兼容性进行搭配!具体可参考我的另外一篇文章

TDD-net跑实验_HITSZ阿星的博客-CSDN博客

3、不仅要修改作者Readme里的checkpoint,还要修改cfgs.py文件里的一些目录。

4、如果跑完实验发现没结果,很可能是没加载成功pcb_30000model.ckpt导致的,具体来说要么是checkpoint地址错,要么是cfgs里地址错

就我而言,我是cfgs文件里

具体来说,在build_whole_network.py里

checkpoint_path = tf.train.latest_checkpoint(os.path.join(cfgs.TRAINED_CKPT, cfgs.VERSION)

cfgs.TRAINED_CKPT = rootpath + /output/trained_weights
cfgs.VERSION = FPN_Res101_0117_OHEM

#也就是说,cfgs.py里的VERSION定义决定了模型加载目录

因此需要在trained_weights里新建文件夹 FPN_Res101_0117_OHEM,然后把那些checkpoint、pcb_30000model.ckpt文件丢进去,然后记得修改checkpoint指向!

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/langs/790056.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-05-05
下一篇2022-05-05

发表评论

登录后才能评论

评论列表(0条)

    保存