Skip to content

Commit 8bf8dd7

Browse files
committed
add pytorch yolov5
1 parent 936ed53 commit 8bf8dd7

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

84 files changed

+13088
-0
lines changed

yolov5/.dockerignore

Lines changed: 220 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,220 @@
1+
# Repo-specific DockerIgnore -------------------------------------------------------------------------------------------
2+
#.git
3+
.cache
4+
.idea
5+
runs
6+
output
7+
coco
8+
storage.googleapis.com
9+
10+
data/samples/*
11+
**/results*.csv
12+
*.jpg
13+
14+
# Neural Network weights -----------------------------------------------------------------------------------------------
15+
**/*.pt
16+
**/*.pth
17+
**/*.onnx
18+
**/*.mlmodel
19+
**/*.torchscript
20+
**/*.torchscript.pt
21+
**/*.tflite
22+
**/*.h5
23+
**/*.pb
24+
*_saved_model/
25+
*_web_model/
26+
27+
# Below Copied From .gitignore -----------------------------------------------------------------------------------------
28+
# Below Copied From .gitignore -----------------------------------------------------------------------------------------
29+
30+
31+
# GitHub Python GitIgnore ----------------------------------------------------------------------------------------------
32+
# Byte-compiled / optimized / DLL files
33+
__pycache__/
34+
*.py[cod]
35+
*$py.class
36+
37+
# C extensions
38+
*.so
39+
40+
# Distribution / packaging
41+
.Python
42+
env/
43+
build/
44+
develop-eggs/
45+
dist/
46+
downloads/
47+
eggs/
48+
.eggs/
49+
lib/
50+
lib64/
51+
parts/
52+
sdist/
53+
var/
54+
wheels/
55+
*.egg-info/
56+
wandb/
57+
.installed.cfg
58+
*.egg
59+
60+
# PyInstaller
61+
# Usually these files are written by a python script from a template
62+
# before PyInstaller builds the exe, so as to inject date/other infos into it.
63+
*.manifest
64+
*.spec
65+
66+
# Installer logs
67+
pip-log.txt
68+
pip-delete-this-directory.txt
69+
70+
# Unit test / coverage reports
71+
htmlcov/
72+
.tox/
73+
.coverage
74+
.coverage.*
75+
.cache
76+
nosetests.xml
77+
coverage.xml
78+
*.cover
79+
.hypothesis/
80+
81+
# Translations
82+
*.mo
83+
*.pot
84+
85+
# Django stuff:
86+
*.log
87+
local_settings.py
88+
89+
# Flask stuff:
90+
instance/
91+
.webassets-cache
92+
93+
# Scrapy stuff:
94+
.scrapy
95+
96+
# Sphinx documentation
97+
docs/_build/
98+
99+
# PyBuilder
100+
target/
101+
102+
# Jupyter Notebook
103+
.ipynb_checkpoints
104+
105+
# pyenv
106+
.python-version
107+
108+
# celery beat schedule file
109+
celerybeat-schedule
110+
111+
# SageMath parsed files
112+
*.sage.py
113+
114+
# dotenv
115+
.env
116+
117+
# virtualenv
118+
.venv*
119+
venv*/
120+
ENV*/
121+
122+
# Spyder project settings
123+
.spyderproject
124+
.spyproject
125+
126+
# Rope project settings
127+
.ropeproject
128+
129+
# mkdocs documentation
130+
/site
131+
132+
# mypy
133+
.mypy_cache/
134+
135+
136+
# https://github.com/github/gitignore/blob/master/Global/macOS.gitignore -----------------------------------------------
137+
138+
# General
139+
.DS_Store
140+
.AppleDouble
141+
.LSOverride
142+
143+
# Icon must end with two \r
144+
Icon
145+
Icon?
146+
147+
# Thumbnails
148+
._*
149+
150+
# Files that might appear in the root of a volume
151+
.DocumentRevisions-V100
152+
.fseventsd
153+
.Spotlight-V100
154+
.TemporaryItems
155+
.Trashes
156+
.VolumeIcon.icns
157+
.com.apple.timemachine.donotpresent
158+
159+
# Directories potentially created on remote AFP share
160+
.AppleDB
161+
.AppleDesktop
162+
Network Trash Folder
163+
Temporary Items
164+
.apdisk
165+
166+
167+
# https://github.com/github/gitignore/blob/master/Global/JetBrains.gitignore
168+
# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and WebStorm
169+
# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
170+
171+
# User-specific stuff:
172+
.idea/*
173+
.idea/**/workspace.xml
174+
.idea/**/tasks.xml
175+
.idea/dictionaries
176+
.html # Bokeh Plots
177+
.pg # TensorFlow Frozen Graphs
178+
.avi # videos
179+
180+
# Sensitive or high-churn files:
181+
.idea/**/dataSources/
182+
.idea/**/dataSources.ids
183+
.idea/**/dataSources.local.xml
184+
.idea/**/sqlDataSources.xml
185+
.idea/**/dynamic.xml
186+
.idea/**/uiDesigner.xml
187+
188+
# Gradle:
189+
.idea/**/gradle.xml
190+
.idea/**/libraries
191+
192+
# CMake
193+
cmake-build-debug/
194+
cmake-build-release/
195+
196+
# Mongo Explorer plugin:
197+
.idea/**/mongoSettings.xml
198+
199+
## File-based project format:
200+
*.iws
201+
202+
## Plugin-specific files:
203+
204+
# IntelliJ
205+
out/
206+
207+
# mpeltonen/sbt-idea plugin
208+
.idea_modules/
209+
210+
# JIRA plugin
211+
atlassian-ide-plugin.xml
212+
213+
# Cursive Clojure plugin
214+
.idea/replstate.xml
215+
216+
# Crashlytics plugin (for Android Studio and IntelliJ)
217+
com_crashlytics_export_strings.xml
218+
crashlytics.properties
219+
crashlytics-build.properties
220+
fabric.properties

yolov5/.gitattributes

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
# this drop notebooks from GitHub language stats
2+
*.ipynb linguist-vendored

yolov5/.github/FUNDING.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
# These are supported funding model platforms
2+
3+
github: glenn-jocher
4+
patreon: ultralytics
5+
open_collective: ultralytics
Lines changed: 85 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,85 @@
1+
name: 🐛 Bug Report
2+
# title: " "
3+
description: Problems with YOLOv5
4+
labels: [bug, triage]
5+
body:
6+
- type: markdown
7+
attributes:
8+
value: |
9+
Thank you for submitting a YOLOv5 🐛 Bug Report!
10+
11+
- type: checkboxes
12+
attributes:
13+
label: Search before asking
14+
description: >
15+
Please search the [issues](https://github.com/ultralytics/yolov5/issues) to see if a similar bug report already exists.
16+
options:
17+
- label: >
18+
I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and found no similar bug report.
19+
required: true
20+
21+
- type: dropdown
22+
attributes:
23+
label: YOLOv5 Component
24+
description: |
25+
Please select the part of YOLOv5 where you found the bug.
26+
multiple: true
27+
options:
28+
- "Training"
29+
- "Validation"
30+
- "Detection"
31+
- "Export"
32+
- "PyTorch Hub"
33+
- "Multi-GPU"
34+
- "Evolution"
35+
- "Integrations"
36+
- "Other"
37+
validations:
38+
required: false
39+
40+
- type: textarea
41+
attributes:
42+
label: Bug
43+
description: Provide console output with error messages and/or screenshots of the bug.
44+
placeholder: |
45+
💡 ProTip! Include as much information as possible (screenshots, logs, tracebacks etc.) to receive the most helpful response.
46+
validations:
47+
required: true
48+
49+
- type: textarea
50+
attributes:
51+
label: Environment
52+
description: Please specify the software and hardware you used to produce the bug.
53+
placeholder: |
54+
- YOLO: YOLOv5 🚀 v6.0-67-g60e42e1 torch 1.9.0+cu111 CUDA:0 (A100-SXM4-40GB, 40536MiB)
55+
- OS: Ubuntu 20.04
56+
- Python: 3.9.0
57+
validations:
58+
required: false
59+
60+
- type: textarea
61+
attributes:
62+
label: Minimal Reproducible Example
63+
description: >
64+
When asking a question, people will be better able to provide help if you provide code that they can easily understand and use to **reproduce** the problem.
65+
This is referred to by community members as creating a [minimal reproducible example](https://stackoverflow.com/help/minimal-reproducible-example).
66+
placeholder: |
67+
```
68+
# Code to reproduce your issue here
69+
```
70+
validations:
71+
required: false
72+
73+
- type: textarea
74+
attributes:
75+
label: Additional
76+
description: Anything else you would like to share?
77+
78+
- type: checkboxes
79+
attributes:
80+
label: Are you willing to submit a PR?
81+
description: >
82+
(Optional) We encourage you to submit a [Pull Request](https://github.com/ultralytics/yolov5/pulls) (PR) to help improve YOLOv5 for everyone, especially if you have a good understanding of how to implement a fix or feature.
83+
See the YOLOv5 [Contributing Guide](https://github.com/ultralytics/yolov5/blob/master/CONTRIBUTING.md) to get started.
84+
options:
85+
- label: Yes I'd like to help by submitting a PR!
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
blank_issues_enabled: true
2+
contact_links:
3+
- name: Slack
4+
url: https://join.slack.com/t/ultralytics/shared_invite/zt-w29ei8bp-jczz7QYUmDtgo6r6KcMIAg
5+
about: Ask on Ultralytics Slack Forum
6+
- name: Stack Overflow
7+
url: https://stackoverflow.com/search?q=YOLOv5
8+
about: Ask on Stack Overflow with 'YOLOv5' tag
Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,50 @@
1+
name: 🚀 Feature Request
2+
description: Suggest a YOLOv5 idea
3+
# title: " "
4+
labels: [enhancement]
5+
body:
6+
- type: markdown
7+
attributes:
8+
value: |
9+
Thank you for submitting a YOLOv5 🚀 Feature Request!
10+
11+
- type: checkboxes
12+
attributes:
13+
label: Search before asking
14+
description: >
15+
Please search the [issues](https://github.com/ultralytics/yolov5/issues) to see if a similar feature request already exists.
16+
options:
17+
- label: >
18+
I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and found no similar feature requests.
19+
required: true
20+
21+
- type: textarea
22+
attributes:
23+
label: Description
24+
description: A short description of your feature.
25+
placeholder: |
26+
What new feature would you like to see in YOLOv5?
27+
validations:
28+
required: true
29+
30+
- type: textarea
31+
attributes:
32+
label: Use case
33+
description: |
34+
Describe the use case of your feature request. It will help us understand and prioritize the feature request.
35+
placeholder: |
36+
How would this feature be used, and who would use it?
37+
38+
- type: textarea
39+
attributes:
40+
label: Additional
41+
description: Anything else you would like to share?
42+
43+
- type: checkboxes
44+
attributes:
45+
label: Are you willing to submit a PR?
46+
description: >
47+
(Optional) We encourage you to submit a [Pull Request](https://github.com/ultralytics/yolov5/pulls) (PR) to help improve YOLOv5 for everyone, especially if you have a good understanding of how to implement a fix or feature.
48+
See the YOLOv5 [Contributing Guide](https://github.com/ultralytics/yolov5/blob/master/CONTRIBUTING.md) to get started.
49+
options:
50+
- label: Yes I'd like to help by submitting a PR!

0 commit comments

Comments
 (0)