tensorflow v2 object detection ValueError 问题

tensorflow v2 object detection ValueError 问题

背景

近期项目需求,再次上手tf的API,顺便升级到了V2。

Windows10 下训练时发现报了这么个错误:

Traceback (most recent call last):
  File "D:\Anaconda3\lib\site-packages\absl\app.py", line 299, in run
    _run_main(main, args)
  File "D:\Anaconda3\lib\site-packages\absl\app.py", line 250, in _run_main
    sys.exit(main(argv))
  File "E:/datasets/DCMCIS/rubbish/model_main_tf2.py", line 103, in main
    use_tpu=FLAGS.use_tpu)
  File "D:\Anaconda3\lib\site-packages\object_detection\model_lib_v2.py", line 554, in train_loop
    unpad_groundtruth_tensors)
  File "D:\Anaconda3\lib\site-packages\object_detection\model_lib_v2.py", line 335, in load_fine_tune_checkpoint
    if not is_object_based_checkpoint(checkpoint_path):
  File "D:\Anaconda3\lib\site-packages\object_detection\model_lib_v2.py", line 298, in is_object_based_checkpoint
    var_names = [var[0] for var in tf.train.list_variables(checkpoint_path)]
  File "D:\Anaconda3\lib\site-packages\tensorflow\python\training\checkpoint_utils.py", line 98, in list_variables
    reader = load_checkpoint(ckpt_dir_or_file)
  File "D:\Anaconda3\lib\site-packages\tensorflow\python\training\checkpoint_utils.py", line 67, in load_checkpoint
    return py_checkpoint_reader.NewCheckpointReader(filename)
  File "D:\Anaconda3\lib\site-packages\tensorflow\python\training\py_checkpoint_reader.py", line 95, in NewCheckpointReader
    return CheckpointReader(compat.as_bytes(filepattern))
ValueError

解决方法

修改config文件,将fine_tune_checkpoint中的\全部改为/

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值