PIDNet
PIDNet copied to clipboard
I Cant train
Hello, I put some of my photos in data/cityscapes/images/train, data/cityscapes/images/val, data/cityscapes/images/train, data/cityscapes/label/train_labels and data/cityscapes/val_labels.
(Absolute path to png = D:\GraduateWork\PIDNet\data\cityscapes\images\train\0010.png,
to its label = D:\GraduateWork\PIDNet\data\cityscapes\labels\train_labels\0010.png, and so on)
Then, i did some lst files: train.lst and val.lst
But when i try to start the train, i get an error: NotImplementedError: Got <class 'NoneType'>, but expected numpy array or torch tensor.
As far as i understand, the error is in the files .lst, im most likely pointing the wrong path's. But I don't understand what exactly I'm doing wrong.
I tried full paths, for example 1 row of train.lst file:
D:\GraduateWork\PIDNet\data\cityscapes\images\train\0010.png D:\GraduateWork\PIDNet\data\cityscapes\labels\train_labels\0010.png
tried paths, starts from a folder data/: cityscapes/images/train/0010.png cityscapes/labels/train_labels/0010.png
Or including it: data/cityscapes/images/train/0010.png data/cityscapes/labels/train_labels/0010.png
Tell me please what am I doing wrong? How to correctly write paths to images in .lst files?
PS im using pidnet_large_cityscapes.yaml , everything inside is right: DATASET: DATASET: cityscapes ROOT: data/ TEST_SET: 'list/cityscapes/val.lst' TRAIN_SET: 'list/cityscapes/train.lst' NUM_CLASSES: 19
I think you should not include the 'DATASET.ROOT' and 'DATASET.DATASET' in the list of paths. So in your example, it should be like this:
images\train\0010.png labels\train_labels\0010.png
I hope it helps