#!/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