sepinf-inc / IPED

IPED Digital Forensic Tool. It is an open source software that can be used to process and analyze digital evidence, often seized at crime scenes by law enforcement or in a corporate investigation by private examiners.
Other
970 stars 220 forks source link

Age estimation for faces #377

Open lfcnassif opened 3 years ago

lfcnassif commented 3 years ago

After testing, improving and merging #361, we can store faces bounding boxes and run some age estimation algorithm like https://www.pyimagesearch.com/2020/04/13/opencv-age-detection-with-deep-learning/

The ImageViewer could also be extended to draw faces bounding boxes stored in items and display ages.

This will be very useful for CSAM cases.

hauck-jvsh commented 3 years ago

This will be great.

lfcnassif commented 3 years ago

some references: https://github.com/diovisgood/agender

lfcnassif commented 3 years ago

Dataset for testing: https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/

lfcnassif commented 3 years ago

Dataset for testing: https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/

Actually, most models are trained using this dataset. If I find some with lots of children, I will let you know.

hauck-jvsh commented 3 years ago

I run some preliminary tests. For the models 4 and 9, results were considered correct if they are within a 3 year error. As for the model 6 the result is considered correct if the age falls into the predict interval. I run these models over a 1000 images of the IMDb data-set and the results were:

Model Correct Incorret
Model 9 460 560
Model 4 229 771
Model 6 162 838
lfcnassif commented 3 years ago

Well, that is worse than I expected. Could you measure the results using precision x recall? I think we should target high recall values, considering CSAM cases.

lfcnassif commented 3 years ago

Maybe @joaomacedo could give some advice here.

lfcnassif commented 2 years ago

Removing the assignee as this is frozen.