pytorch/vision

Pre-trained detection models for PyTorch Hub

Open

#1,945 opened on 2020幎3月6日

GitHub で芋る
 (12 comments) (0 reactions) (0 assignees)Python (6,858 forks)batch import
enhancementhelp wantedmodule: hubmodule: models

Repository metrics

Stars
 (15,050 stars)
PR merge metrics
 (平均マヌゞ 12d 8h) (30d で 14 merged PRs)

説明

🚀 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

コントリビュヌタヌガむド