【发布时间】:2020-12-06 21:46:18
【问题描述】:
我正在尝试找到一个预训练模型,该模型将根据照片中是否有人对图像进行分类。
【问题讨论】:
标签: conv-neural-network pre-trained-model image-classification
我正在尝试找到一个预训练模型,该模型将根据照片中是否有人对图像进行分类。
【问题讨论】:
标签: conv-neural-network pre-trained-model image-classification
您可以为此使用在 COCO 数据集上训练的模型。
例如,对于 Pytorch,您可以查看有关提供的模型的官方文档here。
如果你给它一个简单的搜索 Pytorch 和其他框架的模型,就会有更多种类的模型。
如果您需要有关数据集及其支持的任务的更多信息,可以查看COCO homepage。
您可能还会发现这些有用:
您还可以利用 Detectorn2、mmdetection 等框架来完成这些任务。(或 Tensorflow 的 ObjectDetectionAPI 等)
【讨论】: