model_main.py 无法训练 mobilenet ssd v2 - tensorflow

Question

我正在使用 TensorFlow 1.15 并尝试使用 TensorFlow 对象检测 API 和我自己的数据集来微调 mobilenetSSDv2。

我按照 tf repo here中所述的方式创建了我的 tf 记录，并像这样阅读图像

with tf.gfile.GFile(folder_path+"temp.jpeg", 'rb') as fid:

    encoded_image_data = fid.read()

我已经根据需要将我的分数除以宽度和高度，然后我调整了配置以适应我的班级数量但是当我运行火车过程时我仍然遇到这个错误（我尝试了很多东西都没有成功）

    ...

    

    ...

    

    tensorflow.python.framework.errors_impl.InvalidArgumentError: {{function_node Dataset_map_transform_and_pad_input_data_fn_423}} assertion failed: [[0.576413691][0.335303724][0.766369045]...] [[0.155026451][0.439418][0.299206346]...]     [[{{node Assert/AssertGuard/Assert}}]]      [[IteratorGetNext]]

    

    During handling of the above exception, another exception occurred:

    

    Traceback (most recent call last):   File "./object_detection/model_main.py", line 108, in <module>

        tf.app.run()   File "/home/mai/anaconda3/envs/tf/lib/python3.6/site-packages/tensorflow_core/python/platform/app.py", line 40, in run

        _run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)   File "/home/mai/anaconda3/envs/tf/lib/python3.6/site-packages/absl/app.py", line 299, in run

        _run_main(main, args)   File "/home/mai/anaconda3/envs/tf/lib/python3.6/site-packages/absl/app.py", line 250, in _run_main

        sys.exit(main(argv))   File "./object_detection/model_main.py", line 104, in main

        tf.estimator.train_and_evaluate(estimator, train_spec, eval_specs[0])   File "/home/mai/anaconda3/envs/tf/lib/python3.6/site-packages/tensorflow_estimator/python/estimator/training.py", line 473, in train_and_evaluate

Answer 1

确实是数据，为了修复错误我使用这个repo将我的数据转换为 tf 记录为此需要将数据转换为 YOLO 格式，但这非常简单