config.py

from utils.config import Config
import os

def get_class(classes_path):
    classes_path = os.path.expanduser(classes_path)
    with open(classes_path) as f:
        class_names = f.readlines()
    class_names = [c.strip() for c in class_names]
    return class_names

class CustomerConfig(Config):
    NAME = "Customer"
    NUM_CLASSES = 1 + 1
    RPN_ANCHOR_SCALES = (16, 32, 64, 128, 256)
    # IMAGE_MIN_DIM = 512
    # IMAGE_MAX_DIM = 512
    CLASSES = get_class(r'.\data\coco_classes.txt')
    TRAIN_DATASET = './train_data'
    PRETRAIN_MODEL = "model/mask_rcnn_coco.h5"
    LEARNING_RATE = 1e-5
    EPOCH = 100
    MAX_GT_INSTANCES = 200
    DETECTION_MAX_INSTANCES = 200

class InferenceConfig(Config):
    NAME = 'Customer'
    RPN_ANCHOR_SCALES = (16, 32, 64, 128, 256)
    IMAGE_MIN_DIM = 1024
    IMAGE_MAX_DIM = 1024
    model = r'./model/building_20221121.h5'
    class_path = r'./data/building.names'