search

dpressel / rude-carnie

Age detection in Tensorflow
940 stars 341 forks source link

Using the gender detection model on tf 2.0 #104

Open aayushbhaskar opened 4 years ago

aayushbhaskar commented 4 years ago

Hi, I want to use the gender detection model using tensorflow 2.x. I already have a script to extract faces with a fairly high accuracy, I just want to use the gender detection model using inception v3. Can you please help. I cannot use tensorflow 1.x because my project has many other functionalities, and every other model is written in tf 2.x.

dpressel commented 4 years ago

Not sure if you are trying to run in declarative or eager mode, in 2 eager is default. In general 2.x is a major departure from 1.x and the code will not run even in declarative mode without modifications to use tf.compat.v1 namespaces. If you want to make that change it may just then work. If so, send a PR. I think this tool can probably help you upgrade https://www.tensorflow.org/guide/upgrade

But I haven’t tried it

Otherwise for inference you could export the model for TF serving. I’m not sure your pipeline so maybe it doesn’t make sense here.

In general, this code was developed in 2016 and the TF API has significantly improved since then (including the datasets API and its multiworker handling). So I want to do a major overhaul for 2.x soon but I haven’t had a lot of free cycles ATM and I don’t have a timeline yet.

aayushbhaskar commented 4 years ago

Okay, I may try using the tensorflow code migration method. Do you think the model will still load if the code is changed?