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