plant-seg
plant-seg copied to clipboard
Failed to load or parse config_train.yml: 'loaders'
Hi developer,
I'm trying to train my own data. I checked the tutorial but it is not clear. I followed the tutorial and set the config_train.yml
training: dataset_dir: /projectnb/mylabscc/yzw0093/Train_dataset/Plantseg/data/train/images # contains train/val subdirectories model_name: custom_model in_channels: 1 out_channels: 2 feature_maps: [ 32, 64, 128, 256, 512 ] patch_size: [ 8, 16, 160 ] max_num_iters: 50000 dimensionality: 3D sparse: false device: cuda config_train.yml (END) Besides. Where should I put my mask file and image file?
Thank you so much