pytorch/vision

Pre-trained detection models for PyTorch Hub

Open

#1.945 aberto em 6 de mar. de 2020

Ver no GitHub
 (12 comments) (0 reactions) (0 assignees)Python (6.858 forks)batch import
enhancementhelp wantedmodule: hubmodule: models

Métricas do repositório

Stars
 (15.050 stars)
Métricas de merge de PR
 (Mesclagem média 12d 8h) (14 fundiu PRs em 30d)

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

Guia do colaborador

Direção de pesquisa
Analise o hub conf.py existente para ver como os modelos de classificação são expostos. Adicione pontos de entrada semelhantes para modelos de detecção, por exemplo, fasterrcnn resnet50 fpn. Considere adicionar funções auxiliares para análise de bounding boxes e visualização de detecções. O problema menciona que os blocos de construção já estão lá, então a tarefa principal é conectá los no hub conf.py.
Pilha de tecnologia
pythonpytorch
Domain
machine learningai
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
Meio dia
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PythonPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
60