我正在使用 TensorFlow 1.15 并嘗試使用 TensorFlow 對象檢測 API 和我自己的數(shù)據(jù)集來微調(diào) mobilenetSSDv2。我按照 tf repo here中所述的方式創(chuàng)建了我的 tf 記錄,并像這樣閱讀圖像with tf.gfile.GFile(folder_path+"temp.jpeg", 'rb') as fid: encoded_image_data = fid.read()我已經(jīng)根據(jù)需要將我的分數(shù)除以寬度和高度,然后我調(diào)整了配置以適應(yīng)我的班級數(shù)量但是當我運行火車過程時我仍然遇到這個錯誤(我嘗試了很多東西都沒有成功) ... ... 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
1 回答

蝴蝶不菲
TA貢獻1810條經(jīng)驗 獲得超4個贊
確實是數(shù)據(jù),為了修復(fù)錯誤我使用這個repo將我的數(shù)據(jù)轉(zhuǎn)換為 tf 記錄
為此需要將數(shù)據(jù)轉(zhuǎn)換為 YOLO 格式,但這非常簡單
添加回答
舉報
0/150
提交
取消