model_config_seg.yaml

# Specify things which we want to train multitask netowrk on. Base netowrk perorms object Detection using FasterRCN
# valid types are regression, classification and segmetation and keypoints (all the image points can be specified in keypoints).
# Make sure the dataloder loads all the label names specified by names below

# Background plus Required classes
numclasses: 3
classes: ["color_board","cucumber"] 

# Provide label name, classes and file location with labels for segmentation
segmentation:
    name: ["masks"]
    # classes for which the above segmentation task is for
    classes: [["color_board","cucumber"]]
    file: "data/cucumber/labels_my-project-name_2021-12-17-03-00-40.json"
    

# Provide label name, class, number of classes and file location with labels for classification 
classification:
    name: None
    # For which class object the classification is required
    classes: [None]
    numclasses: [0,0]
    file: "data/cucumber/labels_my-project-name_2021-12-17-03-00-26.csv"

# Provide label name, class and file location with labels for regression
regression:
    name: None
    # For which class the Regression is required
    class: [None]
    file: None

# Provide label name, classes, num of keypoints and file location with labels for keypoint detection
keypoints:
    name: None
    classes: [None]  
    num: 2
    file: "data/cucumber/labels_my-project-name_2021-12-17-03-00-26.csv"