PIDNet icon indicating copy to clipboard operation
PIDNet copied to clipboard

I Cant train

Open AntonDubrovin opened this issue 2 years ago • 1 comments

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) image

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. image

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 image

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

AntonDubrovin avatar Dec 14 '22 20:12 AntonDubrovin

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

oguzhanilter avatar Dec 23 '22 23:12 oguzhanilter