rdev12 / BOSCH-Age-and-Gender-Prediction

Given an input image or video, our model predicts the age (range) and gender for the identified people in each frame. Pipeline consists of super-resolution followed by a multiattribute neural network. Secured 3rd place in BOSCH's event in Inter IIT Tech Meet 10.0.
MIT License
1 stars 0 forks source link

Missing files #1

Open CarmelaCalabrese opened 1 year ago

CarmelaCalabrese commented 1 year ago

Hi! I am trying your code for age detection. I am starting by using Predict.py file but it is looking for checkpoint files that are not available in this repository. Specifically: PATH_TO_CHECKPOINT = 'pedestrian_detection/pretrained_weights/pretrained.pt' PATH_TO_AGE_GENDER_PREDICTOR_CHECKPOINT = 'exp_result/PETA/PETA/img_model/ckpt_max.pth' Could you upload them, please?

Thank you in advance

rdev12 commented 1 year ago

Hey @CarmelaCalabrese, See the updated readme with the link to the GDrive version of the repository which contains all the necessary weights.