modèle de détection d'objets pré-entraîné avec plus de classes que COCO

Question

Pour un projet, j'ai besoin d'un détecteur capable de détecter de nombreux objets différents. Pour cela, les 90 classes de COCO ne suffisent pas car j'aimerais pouvoir en voir plus.
J'ai vu qu'imagenet, par exemple, a beaucoup plus de classes, mais je n'ai pas trouvé de modèle entraîné pour détecter les classes imagenet.
Je programme sur python et je veux éviter de recycler un réseau pour détecter moi-même plus de classes.
J'ai regardé sur pytorch vision et quelques autres référentiels mais je n'ai rien trouvé.
Merci d'avance.

EDIT: J'en ai trouvé un bon maintenant, l'ensemble de données LVIS a 1200 classes pour les détections et utilise les images de coco (ils les ont rebaptisées). Il existe un bon modèle pour cela avec detectron2 de facebookai.https://github.com/facebookresearch/detectron2/blob/master/MODEL_ZOO.md Je pense que ce n'est disponible que pour un environnement cuda (je n'ai pas de GPU :()

Answer 1

Vérifier https://keras.io/api/applications/ pour plus de modèles / ensembles de données

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')

Answer 2

ImageNet n'est malheureusement pas étiqueté à cet effet. Dans le commerce , vous pouvez essayer d'utiliser l' API Tensorflow Object Detection . Il existe des modèles entraînés sur l'ensemble de données OpenImages qui contient 600 classes.

Vous pouvez les utiliser immédiatement pour l'inférence ou si vous le souhaitez, les recycler.