diff --git a/data/config.py b/data/config.py index 91b4c82ea..054b22856 100644 --- a/data/config.py +++ b/data/config.py @@ -172,6 +172,15 @@ def print(self): 'class_names': PASCAL_CLASSES, }) +cig_butts_dataset = dataset_base.copy({ + 'name': 'Immersive Limit - Cigarette Butts', + 'train_info': '/content/drive/MyDrive/Copy of cig_butts.zip (Unzipped Files)/cig_butts/train/coco_annotations.json', + 'train_images': '/content/drive/MyDrive/Copy of cig_butts.zip (Unzipped Files)/cig_butts/train/images/', + 'valid_info': '/content/drive/MyDrive/Copy of cig_butts.zip (Unzipped Files)/cig_butts/val/coco_annotations.json', + 'valid_images': '/content/drive/MyDrive/Copy of cig_butts.zip (Unzipped Files)/cig_butts/val/images/', + 'class_names': ('cig_butt'), + 'label_map': { 1: 1 } +}) @@ -767,6 +776,16 @@ def print(self): }) }) +yolact_resnet50_cig_butts_config = yolact_resnet50_config.copy({ + 'name': 'yolact_plus_resnet50_cig_butts', + # Dataset stuff + 'dataset': cig_butts_dataset, + 'num_classes': len(cig_butts_dataset.class_names) + 1, + + # Image Size + 'max_size': 512, +}) + # ----------------------- YOLACT++ CONFIGS ----------------------- # yolact_plus_base_config = yolact_base_config.copy({