Age-Gender-Estimate-TF
Age-Gender-Estimate-TF copied to clipboard
Problem when using test.py
Hello,
I've tried to execute test.py with the pre-trained model you provide in the description. I am using the following command:
python test.py --images "./data/test" --model_path "./models" --batch_size 128 --cuda
having in ./data/test the tfrecords generated with Wiki dataset and convert_to_records_multiCPU.py, and the pretrained model in models (checkpoint, model.ckpt-14001.data-00000-of-00001 and so on).
However, I am receiving this output:
2018-04-20 13:12:24.750659: I C:\tf_jenkins\home\workspace\rel-win\M\windows\PY\36\tensorflow\core\platform\cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2 !!!TESTING DONE!!! Traceback (most recent call last): File "test.py", line 116, in <module> ckpt.model_checkpoint_path) File "test.py", line 65, in test_once return prob_age_val, real_age, prob_gender_val, real_gender, image_val, np.mean(mean_error_age), np.mean( UnboundLocalError: local variable 'prob_age_val' referenced before assignment
What's going wrong?
Thank you so much!