Tencent / PocketFlow

An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.
https://pocketflow.github.io
Other
2.78k stars 490 forks source link

IndexError: list index out of range #322

Open CoderGenJ opened 4 years ago

CoderGenJ commented 4 years ago

Describe the bug Background: The ssd-vgg-16 model is pruned by learner chn-pruned-rmt method and have been fine trained 120000 iters.It makes a dir named models_cpr_eval to strore the final pruned-trained models. BUG: When i used the export_pb_tflite_models.py script to export a ssd-vgg-16 model to pb file.I get the IndexError. command: python export_pb_tflite_models.py --model_dir=../../models_cpr_eval --input_coll=images_final --output_coll=logits_final Bug log: """Traceback (most recent call last): File "export_pb_tflite_models.py", line 392, in tf.app.run() File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/platform/app.py", line 125, in run _sys.exit(main(argv)) File "export_pb_tflite_models.py", line 371, in main input_name, input_shape = get_input_name_n_shape(meta_path) File "export_pb_tflite_models.py", line 82, in get_input_name_n_shape net_input = tf.get_collection(FLAGS.input_coll)[0] IndexError: list index out of range """ Could you please tell me how to solve this problem?Thanks you in advance.

diligent321 commented 4 years ago

Hi, did you solved it ? @SLAMgreen

I meet the save error, appreciated your kind reply !

gyx1qaz commented 4 years ago

Describe the bug Background: The ssd-vgg-16 model is pruned by learner chn-pruned-rmt method and have been fine trained 120000 iters.It makes a dir named models_cpr_eval to strore the final pruned-trained models. BUG: When i used the export_pb_tflite_models.py script to export a ssd-vgg-16 model to pb file.I get the IndexError. command: python export_pb_tflite_models.py --model_dir=../../models_cpr_eval --input_coll=images_final --output_coll=logits_final Bug log: """Traceback (most recent call last): File "export_pb_tflite_models.py", line 392, in tf.app.run() File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/platform/app.py", line 125, in run _sys.exit(main(argv)) File "export_pb_tflite_models.py", line 371, in main input_name, input_shape = get_input_name_n_shape(meta_path) File "export_pb_tflite_models.py", line 82, in get_input_name_n_shape net_input = tf.get_collection(FLAGS.input_coll)[0] IndexError: list index out of range """ Could you please tell me how to solve this problem?Thanks you in advance.

Describe the bug Background: The ssd-vgg-16 model is pruned by learner chn-pruned-rmt method and have been fine trained 120000 iters.It makes a dir named models_cpr_eval to strore the final pruned-trained models. BUG: When i used the export_pb_tflite_models.py script to export a ssd-vgg-16 model to pb file.I get the IndexError. command: python export_pb_tflite_models.py --model_dir=../../models_cpr_eval --input_coll=images_final --output_coll=logits_final Bug log: """Traceback (most recent call last): File "export_pb_tflite_models.py", line 392, in tf.app.run() File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/platform/app.py", line 125, in run _sys.exit(main(argv)) File "export_pb_tflite_models.py", line 371, in main input_name, input_shape = get_input_name_n_shape(meta_path) File "export_pb_tflite_models.py", line 82, in get_input_name_n_shape net_input = tf.get_collection(FLAGS.input_coll)[0] IndexError: list index out of range """ Could you please tell me how to solve this problem?Thanks you in advance.

gyx1qaz commented 4 years ago

解决了吗?