0

我正在尝试找到一个预先训练的模型,该模型将根据照片中是否有人对图像进行分类。

4

1 回答 1

0

您可以为此使用在数据集上训练的模型COCO
例如,对于 Pytorch,您可以在此处查看有关提供的模型的官方文档。
如果你给它一个简单的搜索 Pytorch 和其他框架的模型,就会有更多种类的模型。

如果您需要有关数据集及其支持的任务的更多信息,可以查看COCO 主页。

您可能还会发现这些很有用:

您还可以利用Detectorn2mmdetection等框架来完成这些任务。(或 Tensorflow 的ObjectDetectionAPI等)

于 2020-12-07T04:46:51.643 回答