search

dsabarinathan / attribute-recognition

This repository contains a PyTorch implementation of a person attribute recognition model. The model has been trained to recognize various attributes such as age, gender, hair length, upper body features, lower body features, and accessories.
MIT License
6 stars 0 forks source link

Train script #1

Closed shekarneo closed 6 months ago

shekarneo commented 7 months ago

Can you provide some information like Dataset and how did you train

dsabarinathan commented 6 months ago

@shekarneo , I used PEDESTRIAN ATTRIBUTE RECOGNITION - WACV'24 dataset for training the model. The dataset contains Market1501 , PA100k and PETA . The train dataset size is 97669 and validation dataset size is 33407