RE_UPLOAD-REBUILD-RESTART

model/layout-model-training/scripts/train_prima.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+
+cd ../tools
+
+python convert_prima_to_coco.py \
+    --prima_datapath ../data/prima \
+    --anno_savepath ../data/prima/annotations.json 
+
+python train_net.py \
+    --dataset_name          prima-layout \
+    --json_annotation_train ../data/prima/annotations-train.json \
+    --image_path_train      ../data/prima/Images \
+    --json_annotation_val   ../data/prima/annotations-val.json \
+    --image_path_val        ../data/prima/Images \
+    --config-file           ../configs/prima/mask_rcnn_R_50_FPN_3x.yaml \
+    OUTPUT_DIR  ../outputs/prima/mask_rcnn_R_50_FPN_3x/ \
+    SOLVER.IMS_PER_BATCH 2