search

dangweili / pedestrian-attribute-recognition-pytorch

A simple baseline for pedestrian attribute recognition in surveillance scenarios
334 stars 81 forks source link

chose the gender and upper and lower color attributes of the rap dataset #11

Open kclch opened 5 years ago

kclch commented 5 years ago

image Hello, I chose the gender and upper and lower color attributes of the rap dataset to train, but the above error occurred. Why is this? Thank you

dangweili commented 5 years ago

dataset['selected_attribute'] should be a list of integers. Maybe you didn't make the right assignments.