pytorch/vision

Pre-trained detection models for PyTorch Hub

Open

#1,945 建立於 2020年3月6日

在 GitHub 查看
 (12 留言) (0 反應) (0 負責人)Python (6,858 fork)batch import
enhancementhelp wantedmodule: hubmodule: models

倉庫指標

Star
 (15,050 star)
PR 合併指標
 (平均合併 12天 8小時) (30 天內合併 14 個 PR)

描述

🚀 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

貢獻者指南

研究方向
查看現有的 hub conf.py，了解分類模型是如何暴露的。為檢測模型添加類似的入口點，例如 fasterrcnn resnet50 fpn。考慮添加用於邊界框解析和視覺化檢測結果的輔助函數。該問題提到建構模組已經存在，因此主要任務是在 hub conf.py 中進行連接。
技術棧
pythonpytorch
領域
machine learningai
議題類型
功能
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
半天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
PythonPyTorch
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
60