-
Notifications
You must be signed in to change notification settings - Fork 15
/
Copy pathExport Mask RCNN SavedModel for inference
49 lines (31 loc) · 1.55 KB
/
Export Mask RCNN SavedModel for inference
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
####################### Export frozen model ##########################
######################################################################################################
Linux command
######################################################################################################
# inside models/research/object_detection
# Path to config file
PIPELINE_CONFIG_PATH = ''
# Path to where your checkpoints were saved
MODEL_DIR = ''
# Path to where the generated frozen model will be saved
OUTPUT_DIR = ''
python exporter_main_v2.py \
--input_type="image_tensor" \
--pipeline_config_path=/path/to/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8-cloud.config \
--trained_checkpoint_dir=/path/to/checkpoints/folder \
--output_directory=/path/to/where/the/exported/model/will/be/saved
######################################################################################################
Windows command
######################################################################################################
# inside models/research/object_detection
# Path to config file
PIPELINE_CONFIG_PATH = ''
# Path to where your checkpoints were saved
MODEL_DIR = ''
# Path to where the generated frozen model will be saved
OUTPUT_DIR = ''
python exporter_main_v2.py ^
--input_type="image_tensor" ^
--pipeline_config_path=/path/to/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8-cloud.config ^
--trained_checkpoint_dir=/path/to/checkpoints/folder ^
--output_directory=/path/to/where/the/exported/model/will/be/saved