search

dangweili / pedestrian-attribute-recognition-pytorch

A simple baseline for pedestrian attribute recognition in surveillance scenarios
332 stars 80 forks source link

modify and select the attributes I want to train #6

Closed kclch closed 5 years ago

kclch commented 5 years ago

I want to select some properties in the attribute dataset for training and testing. How do I modify and select the attributes I need to train in train.prototxt and deploy.prototxt? thank you very much

dangweili commented 5 years ago

Train.txt and deploy.txt don't exist in this version. You can modify selected_attribute in the transform_xxx.py to implement this.

lynnw123 commented 5 years ago

I modified dataset['selected_attribute'] = range(35) to dataset['selected_attribute'] = (0,1,2,3,80) and generated new peta_dataset.pkl, i added --num_att=5 \ in train.sh, but in training, the num_att is 35 instead of 5. I tried modified selected_attribute str in original PETA.mat but with no success, is anything I missed here since I wanted to train the 5 age attributes? thank you very much!

lynnw123 commented 5 years ago

sorry for reopening this, wanted to update that I solved the issue, thanks again!

dangweili commented 5 years ago

@YL-123 I replace the dataset['selected_attribute'] = range(35) to dataset['selected_attribute'] = [0,1,2,3,80], and re-run the transform_peta.py, then everything goes fine. No error happens.

nanhui69 commented 3 years ago

sorry for reopening this, wanted to update that I solved the issue, thanks again!

how do you adreess it ?