-
Notifications
You must be signed in to change notification settings - Fork 15
/
Copy pathRunning training and evaluation of Mask RCNN locally
49 lines (28 loc) · 1.34 KB
/
Running training and evaluation of Mask RCNN locally
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
************************* Linux **********************
######### Training ################
# From the tensorflow/models/research/ directory
python object_detection/model_main_tf2.py \
--pipeline_config_path=/path/to/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8.config \
--model_dir=/path/to/training_process/folder \
--alsologtostderr
########## Evaluation ################
# From the tensorflow/models/research/ directory
python object_detection/model_main_tf2.py \
--pipeline_config_path=/path/to/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8.config \
--model_dir=path/to/training_process/folder \
--checkpoint_dir=path/to/ckpts \
--alsologtostderr
*********************** Windows *******************
######### Training ################
# From the tensorflow/models/research/ directory
python object_detection/model_main_tf2.py ^
--pipeline_config_path=path/to/config/file ^
--model_dir=path/to/training_process/folder ^
--alsologtostderr
########## Evaluation ################
# From the tensorflow/models/research/ directory
python object_detection/model_main_tf2.py ^
--pipeline_config_path=/path/to/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8.config ^
--model_dir=path/to/training_process/folder ^
--checkpoint_dir=path/to/ckpts ^
--alsologtostderr