pytorch/vision

Pre-trained detection models for PyTorch Hub

Open

#1945 aperta il 6 mar 2020

Vedi su GitHub
 (12 commenti) (0 reazioni) (0 assegnatari)Python (6858 fork)batch import
enhancementhelp wantedmodule: hubmodule: models

Metriche repository

Star
 (15.050 star)
Metriche merge PR
 (Merge medio 12g 8h) (14 PR mergiate in 30 g)

Descrizione

🚀 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

Guida contributor

Direzione di ricerca
Esamina l'hub conf.py esistente per vedere come vengono esposti i modelli di classificazione. Aggiungi punti di ingresso simili per i modelli di rilevamento, ad esempio fasterrcnn resnet50 fpn. Considera l'aggiunta di funzioni helper per l'analisi dei bounding box e la visualizzazione dei rilevamenti. Il problema menziona che i blocchi costitutivi sono già presenti, quindi il compito principale è collegarli in hub conf.py.
Tech stack
pythonpytorch
Dominio
machine learningai
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
Mezza giornata
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
PythonPyTorch
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
60