Skip to content

This is a Face Classification project for SJTU machine learning curriculum

Notifications You must be signed in to change notification settings

dzd9798/Machine-Learning-Project

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

机器学习大作业 使用SVM以及卷积神经网络实现人脸分类

/yaleBExtData为所使用的数据集

SVM实现分类代码在/libsvm-3.24/python文件夹下 pr_svm.py为实现分类训练的主代码,使用前需要将代码中的读入路径改为数据集所在路径 visualization.py为特征提取可视化代码

神经网络实现分类代码在/CNN文件夹下 main_cnn.py为实现分类训练的主代码,运行需要tensorflow版本2.0.0 运行main_cnn.py后训练好的模型参数在/CNN/goodmodel文件夹下,可通过CNN_visualization.py读入模型,输出每一层的结果进行可视化操作,改变代码中32行第二个参数的数值即可改变输出的层数。

About

This is a Face Classification project for SJTU machine learning curriculum

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published