tensorflow - 用于“头部姿势”检测的预训练“CNN”模型

Question

亲爱的，

如果你愿意，我正在尝试使用“CNN”实现“头部姿势”（前、上、下、左、右）检测项目，我需要使用预训练的网络，因为我有少量的训练图像。

我的问题是：

• 这种任务的最佳架构是什么（LeNet、VGG、...）？
• 如何决定用我的数据训练哪一层/几层？

注意：我正在使用“Python/Tensorflow/Keras”并且我有一个中档 Nvidia GPU。

非常感谢。

标签： tensorflowkeras