Age estimation in UTKFace using Pytorch
Please follow the steps below to use the facial-age-estimation:
-
Clone this project and Download Dataset: Download the UTKFace dataset from here. unzip folder and Create a new folder named "jpg" within the extracted UTKFace folder. Move all the image files into the "jpg" folder
-
get backbone weight: Run the
swintransformerfacetest.py
. -
get final result: Run the
transformerfacetest_swin_basecontext_8head.py
.
👤 Zero
Give a ⭐️ if this project helped you!