How to get the class labels corresponding to each picture in the validation set

wvinzh / WS_DAN_PyTorch

PyTorch Implementation Of WS-DAN(See Better Before Looking Closer: Weakly Supervised Data Augmentation Network for Fine-Grained Visual Classification)

MIT License

169 stars 34 forks source link

How to get the class labels corresponding to each picture in the validation set #9

Open ke-s opened 3 years ago

ke-s commented 3 years ago

When I did model validation, I got three pictures (raw.jpg,raw_atten.jpg,heat_atten.jpg) I want to get the class label corresponding to each picture in the validation set. What should I do?