File size: 608 Bytes
57b435d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 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 |