Closed Samjith888 closed 4 years ago
Looks like you did not install required library for tensorrt (missing libnvinfer.so.6). Do you need tensorRT? If not, you can remove "--tensorrt=FP32" and your command line will work.
Looks like you did not install required library for tensorrt (missing libnvinfer.so.6). Do you need tensorRT? If not, you can remove "--tensorrt=FP32" and your command line will work.
okay. 2nd query :
I have a trained a model and the checkpoints are saved in a folder 'checkpoints_dir'( after training , there are last 5 checkpoints inside this folder and the best checkpoint is saved in backup folder too). While model conversion into .pb , i have used --ckpt_path=echeckpoints_dir
. So which checkpoint will be used here while converting the model into .pb file using python model_inspect.py
command ?
Hi @Samjith888 , it reads the latest checkpoint in that folder. (or you can see the file with name "checkpoint") You should use the xx/archive as the folder for exporting the best model.
backup folder is useless (I added 'backup' just to deal with rare cases where tf.io.gfile.Gfile may fail for unknown reasons)
hi @mingxingtan , how can we access/save best checkpoints instead of just last 5.
the best checkpint is automatically saved in model_dir/archive/
I have trained a model by using efficientdet-d4, While converting the model , getting following error
command used
python model_inspect.py --runmode=saved_model --model_name=efficientdet-d4 \--ckpt_path=efficientdet-d4/archive/ --saved_model_dir=efficientdet-d4/savedmodeldir \--tensorrt=FP32 --tflite_path=efficientdet-d4/efficientdet-d4.tflite \--hparams=voc_config.yaml
error :
Note : Here the tflite, .pb file and frozen.pb file is generated. Tensorrt file didn't.
1.) Seems the error is with Tensorrt convertion. How to solve this error !!!
2.) Only last 5 checkpoints are saved in the checkpoints directory (none of these 5 checkpoints are not the best checkpoints in my case ) , and the best checkpoints will be saved in 'archive' folder, So i have used '--ckpt_path=efficientdet-d4/archive' as checpoints location for generating .pb file, but this .pb file is not detecting anything while inference. (if i use --ckpt_path=efficientdet-d4/ for generating .pb file then this file is working while inference.)