pytorch/vision

Pre-trained detection models for PyTorch Hub

Open

#1 945 ouverte le 6 mars 2020

Voir sur GitHub
 (12 commentaires) (0 réactions) (0 assignés)Python (6 858 forks)batch import
enhancementhelp wantedmodule: hubmodule: models

Métriques du dépôt

Stars
 (15 050 stars)
Métriques de merge PR
 (Merge moyen 12j 8h) (14 PRs mergées en 30 j)

Description

🚀 Feature

First of all, thanks for the amazing work.

I noticed currently there's no pre-trained detection models exposed via hub_conf.py. It looks like the building blocks are already there. Is there plan to support it?

Thanks

Motivation

Increase coverage of supported vision tasks via Torch Hub.

Pitch

  • entrypoints for detection models are added to hub_conf.py
  • some helpers e.g. for bbox parsing could also be added there

Guide contributeur

Direction de recherche
Examinez le hub conf.py existant pour voir comment les modèles de classification sont exposés. Ajoutez des points d'entrée similaires pour les modèles de détection, par exemple fasterrcnn resnet50 fpn. Envisagez d'ajouter des fonctions d'aide pour l'analyse des boîtes englobantes et la visualisation des détections. Le problème mentionne que les blocs de construction sont déjà présents, donc la tâche principale est de les connecter dans hub conf.py.
Stack technique
pythonpytorch
Domaine
machine learningai
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
Une demi journée
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
PythonPyTorch
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
60