train_cusom.md

Getting Started

Train

• Note: 0 is the background, and the object category starts from 1!
• You need to prepare the data in the format of Dataset Details.
• Modify the configuration file parameters parameter.yaml.
• Modify main.py args's params and model-network-code.

Figure Notes (click to expand)

Test

• Just to see trainval.py

Dataset Details

root：
    images:  
    labels: 
    train.txt：
        /home/dataset/seg/images/train/aaa.jpg
        /home/dataset/seg/images/train/bbb.jpg
    test.txt：
        /home/dataset/seg/images/test/ccc.jpg
    
how to match images and labels?
    '/home/dataset/seg/images/train/aaa.jpg'.replace('images', 'labels')
    or
    '/home/dataset/seg/labels/train/aaa.jpg'.replace('.jpg', '.png')

data enhancement:
    random flip, rotate, crop, noise, 
    hue-brightness-contrast-saturation, zoom(in out), copy-paste?, mosaic?