-
Notifications
You must be signed in to change notification settings - Fork 45
/
convert-model.bat
11 lines (11 loc) · 1.07 KB
/
convert-model.bat
1
2
3
4
5
6
7
8
9
10
11
set OBJECT_DETECTION_PATH=C:\devsbb\projekte\tf-models\research
set PROJECT_PATH=%CD%
set CKPT_NO=16579
set ASDF=c:\my\paths
set TEST=%ASDF%
echo ">>>>>> Cleaning saved model... <<<<<<"
del /S %PROJECT_PATH%\object_detection\saved_model
echo ">>>>>> Exporting saved model... <<<<<<"
python3 %OBJECT_DETECTION_PATH%\object_detection\export_inference_graph.py --input_type=image_tensor --pipeline_config_path=%PROJECT_PATH%\object_detection\faster_rcnn_inception_v2_coco_2018_01_28\pipeline.config --trained_checkpoint_prefix=%PROJECT_PATH%\object_detection\training\model.ckpt-%CKPT_NO% --output_directory=%PROJECT_PATH%\object_detection\saved_model
echo ">>>>>> Converting saved model to tfjs... <<<<<<"
tensorflowjs_converter --input_format=tf_saved_model --output_format=tensorflowjs --saved_model_tags=serve --output_json=true --output_node_names=detection_boxes,detection_classes,detection_features,detection_multiclass_scores,detection_scores,num_detections,raw_detection_boxes,raw_detection_scores %PROJECT_PATH%\object_detection\saved_model\saved_model %PROJECT_PATH%\src\assets\web_model