pytorch/vision

Pre-trained detection models for PyTorch Hub

Open

#1.945 geöffnet am 6. März 2020

Auf GitHub ansehen
 (12 Kommentare) (0 Reaktionen) (0 zugewiesene Personen)Python (6.858 Forks)batch import
enhancementhelp wantedmodule: hubmodule: models

Repository-Metriken

Stars
 (15.050 Stars)
PR-Merge-Metriken
 (Durchschn. Merge 12T 8h) (14 gemergte PRs in 30 T)

Beschreibung

🚀 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

Contributor Guide

Research-Richtung
Schauen Sie sich die vorhandene hub conf.py an, um zu sehen, wie Klassifizierungsmodelle verfügbar gemacht werden. Fügen Sie ähnliche Einstiegspunkte für Erkennungsmodelle hinzu, z. B. fasterrcnn resnet50 fpn. Erwägen Sie, Hilfsfunktionen für die Bounding Box Parsing und Visualisierung von Erkennungen hinzuzufügen. Das Issue erwähnt, dass die Bausteine bereits vorhanden sind, daher besteht die Hauptaufgabe darin, sie in hub conf.py zu verbinden.
Tech Stack
pythonpytorch
Domain
machine learningai
Issue Type
Funktion
SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.
3
Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.
Halber Tag
AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.
Aktiv
KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.
Klar
Voraussetzungen
PythonPyTorch
EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.
60