种族分类器
该代码实现了种族分类器,考虑的种族是:黑人,白人,亚洲人,印度人和西班牙裔。 该模型将任何大小的图像作为输入,并输出图像中人的种族。
训练
python train.py python train_faces.py(提取的人脸模型)在预测中,将模型组合在一起,如果算法设法提取人脸,则使用人脸模型,否则将相互预测。
型号下载
在下载模型然后将其放在称为模型的目录中。
预测
将图像插入到test_data文件夹中,然后运行python Forecast.py对图像进行分类。用于预测的模型是在Imagenet上预训练的VGG16。
Pythonpredict.py --data_dir'/ test / images / folder /'
现在享受您的种族分类器!
2022-04-06 11:12:02
524KB
Python
1