Как точно настроить модель из MTCNN с помощью PyTorch? - PullRequest
0 голосов
/ 06 июня 2019

Я создаю систему, которая может распознавать лица по художественным картинам. Я использую для этого репозиторий face.evoLVe от https://github.com/ZhaoJ9014/face.evoLVe.PyTorch

face.evoLVe использует MTCNN для распознавания лиц и блестяще работает с изображениями реального мира. Тем не менее, я хочу настроить его, используя трансферное обучение, для работы над художественной живописью, такой как Мона Лиза.

Я попробовал учебник на сайте PyTorch, https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html

Однако я не совсем уверен, куда включить модель из face.evoLVe. Так как это MTCNN, я предполагаю, что мне нужно только настроить pnet.npy.

Для набора данных о том, что должно быть точно настроено, я думал о наборе данных с двумя категориями: лица и не лица.

...