python - 比 COCO 具有更多类的预训练对象检测模型

Question

对于一个项目，我需要一个可以检测许多不同物体的检测器。为此，COCO 的 90 个类是不够的，因为我希望能够看到更多。
例如，我已经看到 imagenet 有更多的类，但是我找不到训练来检测 imagenet 类的模型。
我正在用 python 编程，我想避免重新训练网络来自己检测更多的类。
我查看了 pytorch vision 和其他几个存储库，但没有找到任何东西。
提前致谢。

编辑:我现在找到了一个很好的，LVIS 数据集有 1200 个检测类，并且正在使用来自 coco 的图像(他们重新标记了它们)。 facebookai 的detectron2 有很好的模型。 https://github.com/facebookresearch/detectron2/blob/master/MODEL_ZOO.md
我认为它只适用于 cuda 环境(我没有 GPU :()

Answer 1

查询 https://keras.io/api/applications/更多模型/数据集

from tensorflow.keras.applications.resnet50 import ResNet50
from tensorflow.keras.preprocessing import image
from tensorflow.keras.applications.resnet50 import preprocess_input, decode_predictions
import numpy as np

model = ResNet50(weights='imagenet')