train
index
/home/hydang/Downloads/truckdetection-benruelas-master-patch-01932/code_convertted/train.py

 
Modules
       
detectron2.model_zoo
os
posixpath
yaml

 
Functions
       
train(config_file)
Train method for detectron model

 
Data
        config = {'data_getter': {'batch_img_size': 18, 'bottom_right_position': [32.82202, -97.313497], 'output_img_dir': 'dataset/predict_img/dorian_predict/', 'saved_location': 'dataset/saved_location_2_dorian_test2.txt', 'top_left_position': [32.830252, -97.32659], 'zoom_scale': 21}, 'dataset': {'num_workers': 2, 'raw_link': 'https://github.com/trangdao909/TruckDetective/raw/main/TruckDataset_TD.zip', 'test_path': 'truck_test', 'train_path': 'truck_train'}, 'model': {'base_lr': 0.00025, 'ims_per_batch': 2, 'max_iter': 2000, 'model_path': 'COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml', 'num_classes': 1, 'output_csv': 'res/test_result_amazon_warehouse_2.csv', 'output_dir': 'models', 'score_thresh_test': 0.5}}