Traceback (most recent call last):
File "create_tf_record.py", line 155, in <module>
tf.app.run()
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py", line 125, in run
_sys.exit(main(argv))
File "create_tf_record.py", line 151, in main
create_tf_record(root_dir, output_path)
File "create_tf_record.py", line 140, in create_tf_record
img_path, class_label_path, instance_label_path, class_names)
File "create_tf_record.py", line 72, in get_tf_example
cls_lbl = np.argmax(np.bincount(class_label[inst_mask]))
File "/usr/local/lib/python2.7/dist-packages/numpy/core/fromnumeric.py", line 1037, in argmax
return _wrapfunc(a, 'argmax', axis=axis, out=out)
File "/usr/local/lib/python2.7/dist-packages/numpy/core/fromnumeric.py", line 51, in _wrapfunc
return getattr(obj, method)(*args, **kwds)
ValueError: attempt to get argmax of an empty sequence
作成したデータセットを使って学習のスクリプトを実行すると、
のようなエラーが出てしまう。
のようにして確認してみる。
https://gist.github.com/Kanazawanaoaki/53050022fc1a7cfaba87a9c5bb862889
のようにプリント文を入れてlogを見ると、
class_label[inst_mask]
が空になってしまっているデータが一定数ありそれがエラーを起こしているみたい。