weixin_40212554
亨利先生
2019-03-31 09:58

TensorFlow SSD训练自己的数据 checkpoint问题

5
  • tensorflow
  • python
  • 深度学习
  • 人工智能
  • 神经网络

1、参考教程:https://blog.csdn.net/liuyan20062010/article/details/78905517
2、一直到训练成功!
3、导入模型测试,代码

# Restore SSD model.
ckpt_filename = '../train_model/model.ckpt-100'
# ckpt_filename = '../checkpoints/VGG_VOC0712_SSD_300x300_ft_iter_120000.ckpt'
isess.run(tf.global_variables_initializer())
saver = tf.train.Saver()
saver.restore(isess, ckpt_filename)

4、错误信息:

图片说明

5、经网上查阅,说是修改了模型结构的问题,但是只是按教程上面修改了类别数,且训练成功了。其他均未修改。

  • 点赞
  • 回答
  • 收藏
  • 复制链接分享

2条回答

为你推荐

换一换