多标签图像分类的迁移学习

Question

我想构建一个模型，通过两个参数对药丸图像进行分类：shape 和 color。我想使用两个分类器，一个会检测 shape，另一个会检测 color。

有 11 个 color 类和 9 个 shape 类。 我认为使用迁移学习来实现我的目标，例如使用 VGG16 左右，问题是我不太确定如何将 VGG16 与两个分类器一起使用。 我有 10,000 张分类图像，所有图像都有颜色和形状标签，图像为 512x384 RGB。