loliipopshock commited on
Commit
57b435d
·
1 Parent(s): b32ede7

Add the script for prima training

Browse files
Files changed (1) hide show
  1. scripts/train_prima.sh +17 -0
scripts/train_prima.sh ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ cd ../tools
4
+
5
+ python convert_prima_to_coco.py \
6
+ --prima_datapath ../data/prima \
7
+ --anno_savepath ../data/prima/annotations.json
8
+
9
+ python train_net.py \
10
+ --dataset_name prima-layout \
11
+ --json_annotation_train ../data/prima/annotations-train.json \
12
+ --image_path_train ../data/prima/Images \
13
+ --json_annotation_val ../data/prima/annotations-val.json \
14
+ --image_path_val ../data/prima/Images \
15
+ --config-file ../configs/prima/mask_rcnn_R_50_FPN_3x.yaml \
16
+ OUTPUT_DIR ../outputs/prima/mask_rcnn_R_50_FPN_3x/ \
17
+ SOLVER.IMS_PER_BATCH 2