search

Sir-Prise / ahegao-detector

11 stars 2 forks source link

More info about the training #1

Closed Kruk2 closed 3 years ago

Kruk2 commented 3 years ago

Hi! Could you share more information about the model? Could you share your training dataset?

Sir-Prise commented 3 years ago

I got most of the training data by downloading ahegao compilations from YouTube, extracting frames, extracting faces and then manually classifying the pictures.

You can download it here: https://mega.nz/file/KrRWgRhT#fRc2wj0qAUreWrC-tZbgF84ADkF0Nb__CphkzYYkk5Q

This is my first AI experiment, so the layers are just based on a tutorial and I wasn't able to get better results, so I don't post it as it will probably just confuse you :D