#!/usr/bin/env python

import sys
import os
from tools import train as train_mmdet

########################################################################
# config

'''
Usage: 
(1) Use one of the following config files.
(2) Inside the config file, make sure that the dataset that needs to be trained on is uncommented.
(3) Use the appropriate input resolution in the config file (input_size).
(4) Recommend to run the first training with voc0712 dataset as it is widely used and reasonably small.
(5) To convert cityscapes to coco format, run the script: tools/convert_datasets/cityscapes.py

config='./configs/ssd/ssd_mobilenet.py'
config='./configs/ssd/ssd_mobilenet_fpn.py'
config='./configs/ssd/ssd_resnet_fpn.py'
config='./configs/ssd/ssd_regnet_fpn_bgr.py'

config='./configs/retinanet/retinanet_regnet_fpn_bgr.py'
config='./configs/retinanet/retinanet_resnet_fpn.py'
config='./configs/retinanet/fcos_regnet_fpn_bgr.py'
'''

config='./configs/retinanet/retinanet_regnet_fpn_bgr.py'

########################################################################
# other settings
gpus = 4
dataset_style = 'voc' #'voc' #'coco'


########################################################################
# do no run validation for now due to a bug in mmdetection
# https://github.com/open-mmlab/mmdetection/issues/2842
sys.argv = [sys.argv[0], f'--gpus={gpus}', '--no-validate', f'{config}']
# Note: use option to add additional options: eg:
# sys.argv += '--options=resume_from=./work_dirs/ssd512_mobilenet_voc/latest.pth',
args = train_mmdet.parse_args()
train_mmdet.main(args)