Я обучил свой набор данных с помощью ssd_mobilenet_v1_coco_2018_01_28 и получил файлы pipe.config, model.ckpt. Теперь, когда я использую следующую команду,
object_detection/export_tflite_ssd_graph.py \
--pipeline_config_path=$CONFIG_FILE \
--trained_checkpoint_prefix=$CHECKPOINT_PATH \
--output_directory=$OUTPUT_DIR \
--add_postprocessing_op=true
я получаю следующие предупреждения / ошибки:
/home/gpu/anaconda3/envs/vsme/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:523:
FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_qint8 = np.dtype([("qint8", np.int8, 1)])
/home/gpu/anaconda3/envs/vsme/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:524:
FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_quint8 = np.dtype([("quint8", np.uint8, 1)])
/home/gpu/anaconda3/envs/vsme/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:525:
FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_qint16 = np.dtype([("qint16", np.int16, 1)])
/home/gpu/anaconda3/envs/vsme/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:526:
FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_quint16 = np.dtype([("quint16", np.uint16, 1)])
/home/gpu/anaconda3/envs/vsme/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:527:
FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. _np_qint32 = np.dtype([("qint32", np.int32, 1)])
/home/gpu/anaconda3/envs/vsme/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:532:
FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'. np_resource = np.dtype([("resource", np.ubyte, 1)])
Traceback (most recent call last):
File "object_detection/export_tflite_ssd_graph.py", line 143, in <module>
tf.app.run(main)
File "/home/gpu/anaconda3/envs/vsme/lib/python3.6/site-packages/tensorflow/python/platform/app.py", line 125, in run
_sys.exit(main(argv))
File "object_detection/export_tflite_ssd_graph.py", line 139, in main
FLAGS.max_classes_per_detection, use_regular_nms=FLAGS.use_regular_nms)
File "/home/gpu/tfod/models/research/object_detection/export_tflite_ssd_graph_lib.py", line 228, in export_tflite_graph
if image_resizer_config.fixed_shape_resizer.convert_to_grayscale:
AttributeError: convert_to_grayscale