如何使用PyTorch从MTCNN中微调模型?

时间:2019-06-06 17:01:05

标签: conv-neural-network pytorch transfer-learning

我正在开发一种可以检测艺术作品中的面孔的系统。我为此使用https://github.com/ZhaoJ9014/face.evoLVe.PyTorch

中的face.evoLVe回购

face.evoLVe使用MTCNN进行面部检测,并可以在现实世界中出色地工作。但是,我想通过转移学习对诸如蒙娜丽莎的艺术绘画进行微调。

我在PyTorch网站https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html

上尝试了该教程

但是,我不确定在哪里将face.evoLVe中的模型包括在内。由于它是MTCNN,因此我假设我只需要微调pnet.npy。

对于数据集应该进行哪些微调,我正在考虑建立一个具有两类的数据集:面孔和非面孔。

0 个答案:

没有答案