search

tue-robotics / image_recognition

Packages for image recognition - Robocup TU/e Robotics
MIT License
64 stars 41 forks source link

Convert image_recognition_keras/age_gender_recognition to PyTorch #145

Open LoyVanBeek opened 2 years ago

LoyVanBeek commented 2 years ago

We'd like to standardize all machine learning applications at Tech Unitd@Home on PyTorch.

A start has been made in https://github.com/tue-robotics/image_recognition/pull/146

Preferably use the same models, but interface it via PyTorch rather than Keras/Tensorflow/Caffe etc.

LoyVanBeek commented 2 years ago

The model is released at https://github.com/tue-robotics/image_recognition/releases/download/image_recognition_keras_face_properties_weights.28-3.73/image_recognition_keras_face_properties_weights.28-3.73.hdf5