Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update Neuralmagic-YoloV5 integration to v6.1 #29

Merged
merged 763 commits into from
Apr 8, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
763 commits
Select commit Hold shift + click to select a range
a4207a2
Fix TensorRT potential unordered binding addresses (#5826)
imyhxy Nov 30, 2021
bc48457
Handle non-TTY `wandb.errors.UsageError` (#5839)
glenn-jocher Dec 1, 2021
e8f8f2b
Avoid inplace modifying`imgs` in `LoadStreams` (#5850)
passerbythesun Dec 2, 2021
1679aac
Update `LoadImages` `ret_val=False` handling (#5852)
gmt710 Dec 2, 2021
30db14f
Update val.py (#5838)
pradeep-vishnu Dec 2, 2021
00e308f
Update TorchScript suffix to `*.torchscript` (#5856)
glenn-jocher Dec 2, 2021
92a7391
Add `--workers 8` argument to val.py (#5857)
iumyx2612 Dec 2, 2021
360eec6
Update `plot_lr_scheduler()` (#5864)
daikankan Dec 3, 2021
d885799
Update `nl` after `cutout()` (#5873)
glenn-jocher Dec 3, 2021
7bf04d9
`AutoShape()` models as `DetectMultiBackend()` instances (#5845)
glenn-jocher Dec 4, 2021
1075488
Single-command multiple-model export (#5882)
glenn-jocher Dec 4, 2021
fa05f8c
`Detections().tolist()` explicit argument fix (#5907)
lizeng614 Dec 7, 2021
3f152e5
Update wandb_utils.py (#5908)
AyushExel Dec 7, 2021
554f782
Add *.engine (TensorRT extensions) to .gitignore (#5911)
greg2451 Dec 7, 2021
581dc30
Add ONNX inference providers (#5918)
glenn-jocher Dec 8, 2021
7d56d45
Add hardware checks to `notebook_init()` (#5919)
glenn-jocher Dec 8, 2021
a3d5f1d
Revert "Update `plot_lr_scheduler()` (#5864)" (#5920)
glenn-jocher Dec 8, 2021
c77a5a8
Absolute '/content/sample_data' (#5922)
glenn-jocher Dec 8, 2021
5bdb28e
Default PyTorch Hub to `autocast(False)` (#5926)
glenn-jocher Dec 8, 2021
4fb6dd4
Fix ONNX opset inconsistency with parseargs and run args (#5937)
d57montes Dec 9, 2021
c45f9f6
Make `select_device()` robust to `batch_size=-1` (#5940)
youyuxiansen Dec 10, 2021
922fbd8
fix .gitignore not tracking existing folders (#5946)
pasmai Dec 10, 2021
a42af30
Update `strip_optimizer()` (#5949)
iumyx2612 Dec 10, 2021
2c63175
Add nms and agnostic nms to export.py (#5938)
d57montes Dec 10, 2021
8f875d9
Refactor NUM_THREADS (#5954)
glenn-jocher Dec 11, 2021
8f35436
Fix Detections class `tolist()` method (#5945)
yonomitt Dec 11, 2021
19c56e6
Fix `imgsz` bug (#5948)
d57montes Dec 12, 2021
e8ef8fb
`pretrained=False` fix (#5966)
glenn-jocher Dec 13, 2021
2d0c6af
make parameter ignore epochs (#5972)
jinmc Dec 14, 2021
d699c21
YOLOv5s6 params and FLOPs fix (#5977)
glenn-jocher Dec 14, 2021
c9a46a6
Update callbacks.py with `__init__()` (#5979)
glenn-jocher Dec 14, 2021
b7d18f3
Increase `ar_thr` from 20 to 100 for better detection on slender (hig…
MrinalJain17 Dec 15, 2021
da9a1b7
Allow `--weights URL` (#5991)
glenn-jocher Dec 15, 2021
628817d
Recommend `jar xf file.zip` for zips (#5993)
glenn-jocher Dec 15, 2021
c1249a4
*.torchscript inference `self.jit` fix (#6007)
glenn-jocher Dec 16, 2021
407a905
Check TensorRT>=8.0.0 version (#6021)
glenn-jocher Dec 17, 2021
361705d
Multi-layer capable `--freeze` argument (#6019)
youyuxiansen Dec 17, 2021
abbdd48
train -> val comment fix (#6024)
glenn-jocher Dec 17, 2021
26f0415
Add dataset source citations (#6032)
glenn-jocher Dec 19, 2021
0db9d5b
Kaggle `LOGGER` fix (#6041)
glenn-jocher Dec 20, 2021
b8a4bab
Simplify `set_logging()` indexing (#6042)
glenn-jocher Dec 20, 2021
dc54ed5
`--freeze` fix (#6044)
glenn-jocher Dec 20, 2021
95c7bc2
OpenVINO Export (#6057)
glenn-jocher Dec 22, 2021
afa5cfb
Reduce G/D/CIoU logic operations (#6074)
jedi007 Dec 23, 2021
c72270c
Init tensor directly on device (#6068)
deepsworld Dec 23, 2021
db6ec66
W&B: track batch size after autobatch (#6039)
AyushExel Dec 23, 2021
9155eb8
W&B: Log best results after training ends (#6120)
AyushExel Dec 30, 2021
7b6938d
Log best results (#6085)
awsaf49 Dec 30, 2021
affa284
Refactor/reduce G/C/D/IoU `if: else` statements (#6087)
cmoseses Dec 30, 2021
d95978a
Add EdgeTPU support (#3630)
zldrobit Dec 31, 2021
e1dc894
Enable AdamW optimizer (#6152)
bilzard Jan 2, 2022
ec4b6dd
Update export format docstrings (#6151)
glenn-jocher Jan 3, 2022
968e300
Update greetings.yml (#6165)
glenn-jocher Jan 3, 2022
5344e54
[pre-commit.ci] pre-commit suggestions (#6177)
pre-commit-ci[bot] Jan 3, 2022
b4a29b5
Update NMS `max_wh=7680` for 8k images (#6178)
glenn-jocher Jan 3, 2022
63a4d86
Add OpenVINO inference (#6179)
glenn-jocher Jan 3, 2022
7cad659
Ignore `*_openvino_model/` dir (#6180)
glenn-jocher Jan 4, 2022
5bd6a97
Global export format sort (#6182)
glenn-jocher Jan 4, 2022
fb83929
Fix TorchScript on mobile export (#6183)
yinrong Jan 4, 2022
a2f4a17
TensorRT 7 `anchor_grid` compatibility fix (#6185)
imyhxy Jan 4, 2022
7b31a53
Add `tensorrt>=7.0.0` checks (#6193)
glenn-jocher Jan 4, 2022
b5b56a3
Add CoreML inference (#6195)
glenn-jocher Jan 5, 2022
9e9219f
Fix `nan`-robust stream FPS (#6198)
glenn-jocher Jan 5, 2022
5402753
Edge TPU compiler comment (#6196)
glenn-jocher Jan 5, 2022
8125ec5
TFLite `--int8` 'flatbuffers==1.12' fix (#6216)
glenn-jocher Jan 5, 2022
00d7b97
TFLite `--int8` 'flatbuffers==1.12' fix 2 (#6217)
glenn-jocher Jan 5, 2022
b4ac3df
Add `edgetpu_compiler` checks (#6218)
glenn-jocher Jan 5, 2022
f80c463
Attempt `edgetpu-compiler` autoinstall (#6223)
glenn-jocher Jan 6, 2022
ad565e3
Update README speed reproduction command (#6228)
glenn-jocher Jan 6, 2022
33a67b4
Update P2-P7 `models/hub` variants (#6230)
glenn-jocher Jan 6, 2022
6865d19
TensorRT 7 export fix (#6235)
glenn-jocher Jan 7, 2022
9b13a59
Fix `cmd` string on `tfjs` export (#6243)
Jan 10, 2022
b3eaf50
TensorRT pip install
glenn-jocher Jan 11, 2022
f3085ac
Enable ONNX `--half` FP16 inference (#6268)
glenn-jocher Jan 11, 2022
80473a6
Update `export.py` with Detect, Validate usages (#6280)
glenn-jocher Jan 13, 2022
af00134
Add `is_kaggle()` function (#6285)
glenn-jocher Jan 13, 2022
e7bf382
Fix `device` count check (#6290)
glenn-jocher Jan 14, 2022
a1a9c68
Fixing bug multi-gpu training (#6299)
Jan 14, 2022
436ffc4
`select_device()` cleanup (#6302)
glenn-jocher Jan 15, 2022
db1f83b
Fix `train.py` parameter groups desc error (#6318)
Otfot Jan 17, 2022
3119b2f
Remove `dataset_stats()` autodownload capability (#6303)
glenn-jocher Jan 18, 2022
fd55271
Console corrupted -> corrupt (#6338)
glenn-jocher Jan 18, 2022
e2e95b2
TensorRT `assert im.device.type != 'cpu'` on export (#6340)
glenn-jocher Jan 18, 2022
0cf932b
`export.py` return exported files/dirs (#6343)
glenn-jocher Jan 19, 2022
e521909
Created using Colaboratory
glenn-jocher Jan 19, 2022
750c42e
`export.py` automatic `forward_export` (#6352)
glenn-jocher Jan 19, 2022
9708cf5
New environment variable `VERBOSE` (#6353)
johnk2hawaii Jan 20, 2022
4e841b9
Reuse `de_parallel()` rather than `is_parallel()` (#6354)
imyhxy Jan 20, 2022
e1893c8
`DEVICE_COUNT` instead of `WORLD_SIZE` to calculate `nw` (#6324)
sitecao Jan 20, 2022
9bcc32a
Flush callbacks when on `--evolve` (#6374)
AyushExel Jan 20, 2022
1b41a1d
FROM nvcr.io/nvidia/pytorch:21.12-py3 (#6377)
glenn-jocher Jan 21, 2022
bd815d4
FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6379)
glenn-jocher Jan 21, 2022
c43439a
Add `albumentations` to Dockerfile (#6392)
glenn-jocher Jan 22, 2022
8efe977
Add `stop_training=False` flag to callbacks (#6365)
haimat Jan 23, 2022
482af47
Add `detect.py` GIF video inference (#6410)
glenn-jocher Jan 24, 2022
cfecd90
Update `greetings.yaml` email address (#6412)
glenn-jocher Jan 24, 2022
ed9bac8
Rename logger from 'utils.logger' to 'yolov5' (#6421)
JonathanSamelson Jan 25, 2022
16563ac
Prefer `tflite_runtime` for TFLite inference if installed (#6406)
motokimura Jan 25, 2022
ff8646c
Update workflows (#6427)
glenn-jocher Jan 25, 2022
d5966c9
Namespace `VERBOSE` env variable to `YOLOv5_VERBOSE` (#6428)
glenn-jocher Jan 25, 2022
3b7ac28
Add `*.asf` video support (#6436)
toschi23 Jan 26, 2022
fe7de6a
Revert "Remove `dataset_stats()` autodownload capability (#6303)" (#6…
glenn-jocher Jan 26, 2022
856d4e5
Fix `select_device()` for Multi-GPU (#6434)
glenn-jocher Jan 26, 2022
d8b5beb
Fix2 `select_device()` for Multi-GPU (#6461)
glenn-jocher Jan 28, 2022
7539cd7
Add Product Hunt social media icon (#6464)
glenn-jocher Jan 28, 2022
6445a81
Resolve dataset paths (#6489)
glenn-jocher Feb 1, 2022
b884ea3
Simplify TF normalized to pixels (#6494)
glenn-jocher Feb 1, 2022
5e4ff19
Improved `export.py` usage examples (#6495)
glenn-jocher Feb 1, 2022
77977e0
CoreML inference fix `list()` -> `sorted()` (#6496)
glenn-jocher Feb 1, 2022
842d049
Suppress `torch.jit.TracerWarning` on export (#6498)
glenn-jocher Feb 1, 2022
4c40933
Suppress export.run() TracerWarnings (#6499)
glenn-jocher Feb 1, 2022
b73c62e
W&B: Remember batchsize on resuming (#6512)
AyushExel Feb 3, 2022
19e0208
Update hyp.scratch-high.yaml (#6525)
glenn-jocher Feb 3, 2022
cb40c9a
TODO issues exempt from stale action (#6530)
glenn-jocher Feb 3, 2022
c3e599c
Update val_batch*.jpg for Chinese fonts (#6526)
glenn-jocher Feb 3, 2022
a82292e
Social icons after text (#6473)
glenn-jocher Feb 3, 2022
8d05716
Edge TPU compiler `sudo` fix (#6531)
glenn-jocher Feb 3, 2022
9bc72a3
Edge TPU export 'list index out of range' fix (#6533)
glenn-jocher Feb 3, 2022
079b36d
Edge TPU `tf.lite.experimental.load_delegate` fix (#6536)
glenn-jocher Feb 4, 2022
8fcdf3b
Fixing minor multi-streaming issues with TensoRT engine (#6504)
greg2451 Feb 4, 2022
aff0281
Load checkpoint on CPU instead of on GPU (#6516)
bilzard Feb 4, 2022
e1a6a0b
flake8: code meanings (#6481)
Borda Feb 4, 2022
cba4303
Fix 6 Flake8 issues (#6541)
Borda Feb 5, 2022
dc7e093
Edge TPU TF imports fix (#6542)
glenn-jocher Feb 5, 2022
b57abb1
Move trainloader functions to class methods (#6559)
glenn-jocher Feb 7, 2022
006eb40
Improved AutoBatch DDP error message (#6568)
glenn-jocher Feb 8, 2022
22da63e
Fix zero-export handling with `if any(f):` (#6569)
glenn-jocher Feb 8, 2022
f627bc5
Fix `plot_labels()` colored histogram bug (#6574)
glenn-jocher Feb 8, 2022
f40854b
Allow custom` --evolve` project names (#6567)
MattVAD Feb 8, 2022
9c513ca
Add `DATASETS_DIR` global in general.py (#6578)
glenn-jocher Feb 8, 2022
9cf80b7
return `opt` from `train.run()` (#6581)
chf4850 Feb 9, 2022
364b7c5
Fix YouTube dislike button bug in `pafy` package (#6603)
glenn-jocher Feb 10, 2022
a5c9057
Update train.py
glenn-jocher Feb 10, 2022
c21da59
Fix `hyp_evolve.yaml` indexing bug (#6604)
glenn-jocher Feb 10, 2022
cb2ad9f
Fix `ROOT / data` when running W&B `log_dataset()` (#6606)
or-toledano Feb 10, 2022
b40bdd6
YouTube dependency fix `youtube_dl==2020.12.2` (#6612)
glenn-jocher Feb 11, 2022
3eef804
Add YOLOv5n to Reproduce section (#6619)
glenn-jocher Feb 11, 2022
51f7750
W&B: Improve resume stability (#6611)
AyushExel Feb 12, 2022
96d8f86
W&B: don't log media in evolve (#6617)
AyushExel Feb 12, 2022
a45e472
YOLOv5 Export Benchmarks (#6613)
glenn-jocher Feb 12, 2022
5389300
Fix ConfusionMatrix scale `vmin=0.0` (#6638)
glenn-jocher Feb 14, 2022
ca0a007
Fixed wandb logger KeyError (#6637)
imyhxy Feb 14, 2022
ee6c70a
Fix yolov3.yaml remove list (#6655)
glenn-jocher Feb 16, 2022
6769021
Validate with 2x `--workers` (#6658)
glenn-jocher Feb 16, 2022
1ff4370
Validate with 2x `--workers` single-GPU/CPU fix (#6659)
glenn-jocher Feb 16, 2022
7b80545
Add `--cache val` (#6663)
glenn-jocher Feb 16, 2022
2e5c67e
Robust `scipy.cluster.vq.kmeans` too few points (#6668)
glenn-jocher Feb 17, 2022
9a8ebe6
Update Dockerfile `torch==1.10.2+cu113` (#6669)
glenn-jocher Feb 17, 2022
bb054d5
FROM nvcr.io/nvidia/pytorch:22.01-py3 (#6670)
glenn-jocher Feb 17, 2022
09d263b
FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6671)
glenn-jocher Feb 17, 2022
2545628
Update Dockerfile reorder installs (#6672)
glenn-jocher Feb 17, 2022
18b5081
FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6673)
glenn-jocher Feb 17, 2022
07221f1
FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6677)
glenn-jocher Feb 17, 2022
66e5d79
Fix TF exports >= 2GB (#6292)
zldrobit Feb 18, 2022
3670d91
Fix `--evolve --bucket gs://...` (#6698)
glenn-jocher Feb 18, 2022
56697c4
Fix CoreML P6 inference (#6700)
glenn-jocher Feb 18, 2022
0365379
Fix floating point in number of workers `nw` (#6701)
SamuelYvon Feb 18, 2022
a297efc
Edge TPU inference fix (#6686)
RaffaeleGalliera Feb 19, 2022
de9c25b
Use `export_formats()` in export.py (#6705)
glenn-jocher Feb 19, 2022
4de8b24
Suppress `torch` AMP-CPU warnings (#6706)
glenn-jocher Feb 19, 2022
c43f135
Update `nw` to `max(nd, 1)` (#6714)
glenn-jocher Feb 20, 2022
dbbb57c
GH: add PR template (#6482)
Borda Feb 21, 2022
a936f5f
Switch default LR scheduler from cos to linear (#6729)
glenn-jocher Feb 21, 2022
2692e67
Updated VOC hyperparameters (#6732)
glenn-jocher Feb 22, 2022
3752807
YOLOv5 v6.1 release (#6739)
glenn-jocher Feb 22, 2022
66aaf51
Pre-commit table fix (#6744)
glenn-jocher Feb 22, 2022
4bab565
Update tutorial.ipynb (2 CPUs, 12.7 GB RAM, 42.2/166.8 GB disk) (#6767)
glenn-jocher Feb 24, 2022
47f265d
Update min warmup iterations from 1k to 100 (#6768)
glenn-jocher Feb 24, 2022
c161557
Default `OMP_NUM_THREADS=8` (#6770)
glenn-jocher Feb 24, 2022
741cd0e
Update tutorial.ipynb (#6771)
glenn-jocher Feb 24, 2022
cea994b
Update hyp.VOC.yaml (#6772)
glenn-jocher Feb 24, 2022
b2adc7c
Fix export for 1-channel images (#6780)
lcombaldieu Feb 25, 2022
0f81991
Update EMA decay `tau` (#6769)
glenn-jocher Feb 25, 2022
9ec51a6
YOLOv5s6 params FLOPs fix (#6782)
glenn-jocher Feb 25, 2022
c2403eb
Update PULL_REQUEST_TEMPLATE.md (#6783)
glenn-jocher Feb 25, 2022
63ddb6f
Update autoanchor.py (#6794)
glenn-jocher Feb 26, 2022
bcc92e2
Update sweep.yaml (#6825)
lcombaldieu Mar 4, 2022
601dbb8
AutoAnchor improved initialization robustness (#6854)
glenn-jocher Mar 4, 2022
8a66eba
Add `*.ts` to `VID_FORMATS` (#6859)
glenn-jocher Mar 4, 2022
4728840
Update `--cache disk` deprecate `*_npy/` dirs (#6876)
glenn-jocher Mar 6, 2022
7e98b48
Update yolov5s.yaml (#6865)
vnekat Mar 6, 2022
596de6d
Default FP16 TensorRT export (#6798)
DavidBaldsiefen Mar 6, 2022
c8a5899
Bump actions/setup-python from 2 to 3 (#6880)
dependabot[bot] Mar 7, 2022
a5a1760
Bump actions/checkout from 2 to 3 (#6881)
dependabot[bot] Mar 7, 2022
acc58c1
Fix TRT `max_workspace_size` deprecation notice (#6856)
glenn-jocher Mar 7, 2022
e6e36aa
Update bytes to GB with bitshift (#6886)
glenn-jocher Mar 7, 2022
6dd82c0
Move `git_describe()` to general.py (#6918)
glenn-jocher Mar 9, 2022
d3d9cbc
PyTorch 1.11.0 compatibility updates (#6932)
glenn-jocher Mar 10, 2022
055e72a
Optimize PyTorch 1.11.0 compatibility update (#6933)
glenn-jocher Mar 10, 2022
caf7ad0
Allow 3-point segments (#6938)
glenn-jocher Mar 10, 2022
84efa62
Fix PyTorch Hub export inference shapes (#6949)
glenn-jocher Mar 11, 2022
b94b59e
DetectMultiBackend() `--half` handling (#6945)
glenn-jocher Mar 11, 2022
c6b4f84
Update Dockerfile `torch==1.11.0+cu113` (#6954)
glenn-jocher Mar 11, 2022
c84dd27
New val.py `cuda` variable (#6957)
glenn-jocher Mar 12, 2022
52c1399
DetectMultiBackend() return `device` update (#6958)
glenn-jocher Mar 12, 2022
701e117
Tensor initialization on device improvements (#6959)
glenn-jocher Mar 12, 2022
c13d4ce
EdgeTPU optimizations (#6808)
paradigmn Mar 12, 2022
2d45de6
Model `ema` key backward compatibility fix (#6972)
glenn-jocher Mar 14, 2022
99de551
pt model to cpu on TF export
glenn-jocher Mar 14, 2022
932dc78
YOLOv5 Export Benchmarks for GPU (#6963)
glenn-jocher Mar 14, 2022
c09fb2a
Update TQDM bar format (#6988)
glenn-jocher Mar 15, 2022
3f634d4
Conditional `Timeout()` by OS (disable on Windows) (#7013)
glenn-jocher Mar 16, 2022
7c6a335
fix: add default PIL font as fallback (#7010)
maxstrobel Mar 17, 2022
4effd06
Consistent saved_model output format (#7032)
MrinalJain17 Mar 18, 2022
b0ba101
`ComputeLoss()` indexing/speed improvements (#7048)
glenn-jocher Mar 20, 2022
9ebec78
Update Dockerfile to `git clone` instead of `COPY` (#7053)
glenn-jocher Mar 20, 2022
6843ea5
Create SECURITY.md (#7054)
glenn-jocher Mar 20, 2022
9f4d113
Fix incomplete URL substring sanitation (#7056)
glenn-jocher Mar 20, 2022
529fbc1
Use PIL to eliminate chroma subsampling in crops (#7008)
LaserBorg Mar 20, 2022
f327eee
Fix `check_anchor_order()` in pixel-space not grid-space (#7060)
glenn-jocher Mar 20, 2022
d5e363f
Update detect.py non-inplace with `y.tensor_split()` (#7062)
glenn-jocher Mar 20, 2022
0529b77
Update common.py lists for tuples (#7063)
glenn-jocher Mar 20, 2022
e278fd6
Update W&B message to `LOGGER.info()` (#7064)
glenn-jocher Mar 20, 2022
9e75cbf
Update __init__.py (#7065)
glenn-jocher Mar 20, 2022
178c109
Add non-zero `da` `check_anchor_order()` condition (#7066)
glenn-jocher Mar 20, 2022
9cd89b7
Fix2 `check_anchor_order()` in pixel-space not grid-space (#7067)
glenn-jocher Mar 20, 2022
9b771a3
Revert "Update detect.py non-inplace with `y.tensor_split()` (#7062)"…
glenn-jocher Mar 21, 2022
6f12803
Update loss.py with `if self.gr < 1:` (#7087)
glenn-jocher Mar 21, 2022
a2d617e
Update loss for FP16 `tobj` (#7088)
glenn-jocher Mar 21, 2022
a600bae
Update model summary to display model name (#7101)
glenn-jocher Mar 22, 2022
05aae17
`torch.split()` 1.7.0 compatibility fix (#7102)
glenn-jocher Mar 22, 2022
ee0b3b2
Update benchmarks significant digits (#7103)
glenn-jocher Mar 22, 2022
6134ec5
Model summary `pathlib` fix (#7104)
glenn-jocher Mar 22, 2022
ecc2c7b
Remove named arguments where possible (#7105)
glenn-jocher Mar 22, 2022
c3ae4e4
Multi-threaded VisDrone and VOC downloads (#7108)
glenn-jocher Mar 23, 2022
bc3ed95
`np.fromfile()` Chinese image paths fix (#6979)
yeshanliu Mar 23, 2022
a0a4adf
Add PyTorch Hub `results.save(labels=False)` option (#7129)
glenn-jocher Mar 24, 2022
8eaecd2
SparseML integration
KSGulin Mar 25, 2022
1f73552
Add SparseML dependancy
KSGulin Mar 25, 2022
4db2a15
Update: add missing files
KSGulin Mar 26, 2022
70fb4cd
Update requirements.txt
KSGulin Mar 30, 2022
bf225ec
Update: sparseml-nightly support
KSGulin Mar 30, 2022
4b4b932
Merge branch 'update-6.1' of github.com:neuralmagic/yolov5 into updat…
KSGulin Mar 30, 2022
6d3667a
Update: remove model versioning
KSGulin Mar 30, 2022
28579c8
Partial update for multi-stage recipes
KSGulin Apr 5, 2022
e5999d5
Update: multi-stage recipe support
KSGulin Apr 6, 2022
3218a78
Update: remove sparseml dep
KSGulin Apr 6, 2022
bbbcf6b
Fix: multi-stage recipe handeling
KSGulin Apr 6, 2022
140ee49
Fix: multi stage support
KSGulin Apr 7, 2022
912040c
Fix: non-recipe runs
KSGulin Apr 8, 2022
a8dfa0f
Add: legacy hyperparam files
KSGulin Apr 8, 2022
7bdf2e6
Fix: add copy-paste to hyps
KSGulin Apr 8, 2022
20f6f91
Fix: nit
KSGulin Apr 8, 2022
5eadf3a
apply structure fixes
Apr 8, 2022
011e7df
Squashed rebase to v6.1 upstream
Apr 8, 2022
a32b970
Update SparseML Integration to V6.1 (#26)
KSGulin Apr 8, 2022
ddd5851
manager fixes
KSGulin Apr 8, 2022
70aef0b
Merge branch 'update-6.1' of github.com:neuralmagic/yolov5 into updat…
KSGulin Apr 8, 2022
dd1ad41
Update function name
KSGulin Apr 8, 2022
4d686e5
Merge branch 'update-6.1' into upstream-master
KSGulin Apr 8, 2022
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 9 additions & 3 deletions .dockerignore
Original file line number Diff line number Diff line change
Expand Up @@ -8,17 +8,23 @@ coco
storage.googleapis.com

data/samples/*
**/results*.txt
**/results*.csv
*.jpg

# Neural Network weights -----------------------------------------------------------------------------------------------
**/*.weights
**/*.pt
**/*.pth
**/*.onnx
**/*.engine
**/*.mlmodel
**/*.torchscript

**/*.torchscript.pt
**/*.tflite
**/*.h5
**/*.pb
*_saved_model/
*_web_model/
*_openvino_model/

# Below Copied From .gitignore -----------------------------------------------------------------------------------------
# Below Copied From .gitignore -----------------------------------------------------------------------------------------
Expand Down
55 changes: 0 additions & 55 deletions .github/ISSUE_TEMPLATE/bug-report.md

This file was deleted.

85 changes: 85 additions & 0 deletions .github/ISSUE_TEMPLATE/bug-report.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,85 @@
name: 🐛 Bug Report
# title: " "
description: Problems with YOLOv5
labels: [bug, triage]
body:
- type: markdown
attributes:
value: |
Thank you for submitting a YOLOv5 🐛 Bug Report!
- type: checkboxes
attributes:
label: Search before asking
description: >
Please search the [issues](https://github.com/ultralytics/yolov5/issues) to see if a similar bug report already exists.
options:
- label: >
I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and found no similar bug report.
required: true
- type: dropdown
attributes:
label: YOLOv5 Component
description: |
Please select the part of YOLOv5 where you found the bug.
multiple: true
options:
- "Training"
- "Validation"
- "Detection"
- "Export"
- "PyTorch Hub"
- "Multi-GPU"
- "Evolution"
- "Integrations"
- "Other"
validations:
required: false

- type: textarea
attributes:
label: Bug
description: Provide console output with error messages and/or screenshots of the bug.
placeholder: |
💡 ProTip! Include as much information as possible (screenshots, logs, tracebacks etc.) to receive the most helpful response.
validations:
required: true

- type: textarea
attributes:
label: Environment
description: Please specify the software and hardware you used to produce the bug.
placeholder: |
- YOLO: YOLOv5 🚀 v6.0-67-g60e42e1 torch 1.9.0+cu111 CUDA:0 (A100-SXM4-40GB, 40536MiB)
- OS: Ubuntu 20.04
- Python: 3.9.0
validations:
required: false

- type: textarea
attributes:
label: Minimal Reproducible Example
description: >
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.
This is referred to by community members as creating a [minimal reproducible example](https://stackoverflow.com/help/minimal-reproducible-example).
placeholder: |
```
# Code to reproduce your issue here
```
validations:
required: false

- type: textarea
attributes:
label: Additional
description: Anything else you would like to share?

- type: checkboxes
attributes:
label: Are you willing to submit a PR?
description: >
(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.
See the YOLOv5 [Contributing Guide](https://github.com/ultralytics/yolov5/blob/master/CONTRIBUTING.md) to get started.
options:
- label: Yes I'd like to help by submitting a PR!
8 changes: 8 additions & 0 deletions .github/ISSUE_TEMPLATE/config.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
blank_issues_enabled: true
contact_links:
- name: Slack
url: https://join.slack.com/t/ultralytics/shared_invite/zt-w29ei8bp-jczz7QYUmDtgo6r6KcMIAg
about: Ask on Ultralytics Slack Forum
- name: Stack Overflow
url: https://stackoverflow.com/search?q=YOLOv5
about: Ask on Stack Overflow with 'YOLOv5' tag
27 changes: 0 additions & 27 deletions .github/ISSUE_TEMPLATE/feature-request.md

This file was deleted.

50 changes: 50 additions & 0 deletions .github/ISSUE_TEMPLATE/feature-request.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
name: 🚀 Feature Request
description: Suggest a YOLOv5 idea
# title: " "
labels: [enhancement]
body:
- type: markdown
attributes:
value: |
Thank you for submitting a YOLOv5 🚀 Feature Request!

- type: checkboxes
attributes:
label: Search before asking
description: >
Please search the [issues](https://github.com/ultralytics/yolov5/issues) to see if a similar feature request already exists.
options:
- label: >
I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and found no similar feature requests.
required: true

- type: textarea
attributes:
label: Description
description: A short description of your feature.
placeholder: |
What new feature would you like to see in YOLOv5?
validations:
required: true

- type: textarea
attributes:
label: Use case
description: |
Describe the use case of your feature request. It will help us understand and prioritize the feature request.
placeholder: |
How would this feature be used, and who would use it?

- type: textarea
attributes:
label: Additional
description: Anything else you would like to share?

- type: checkboxes
attributes:
label: Are you willing to submit a PR?
description: >
(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.
See the YOLOv5 [Contributing Guide](https://github.com/ultralytics/yolov5/blob/master/CONTRIBUTING.md) to get started.
options:
- label: Yes I'd like to help by submitting a PR!
13 changes: 0 additions & 13 deletions .github/ISSUE_TEMPLATE/question.md

This file was deleted.

33 changes: 33 additions & 0 deletions .github/ISSUE_TEMPLATE/question.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
name: ❓ Question
description: Ask a YOLOv5 question
# title: " "
labels: [question]
body:
- type: markdown
attributes:
value: |
Thank you for asking a YOLOv5 ❓ Question!

- type: checkboxes
attributes:
label: Search before asking
description: >
Please search the [issues](https://github.com/ultralytics/yolov5/issues) and [discussions](https://github.com/ultralytics/yolov5/discussions) to see if a similar question already exists.
options:
- label: >
I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and [discussions](https://github.com/ultralytics/yolov5/discussions) and found no similar questions.
required: true

- type: textarea
attributes:
label: Question
description: What is your question?
placeholder: |
💡 ProTip! Include as much information as possible (screenshots, logs, tracebacks etc.) to receive the most helpful response.
validations:
required: true

- type: textarea
attributes:
label: Additional
description: Anything else you would like to share?
9 changes: 9 additions & 0 deletions .github/PULL_REQUEST_TEMPLATE.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
<!--
Thank you for submitting a YOLOv5 🚀 Pull Request! We want to make contributing to YOLOv5 as easy and transparent as possible. A few tips to get you started:

- Search existing YOLOv5 [PRs](https://github.com/ultralytics/yolov5/pull) to see if a similar PR already exists.
- Link this PR to a YOLOv5 [issue](https://github.com/ultralytics/yolov5/issues) to help us understand what bug fix or feature is being implemented.
- Provide before and after profiling/inference/training results to help us quantify the improvement your PR provides (if applicable).

Please see our ✅ [Contributing Guide](https://github.com/ultralytics/yolov5/blob/master/CONTRIBUTING.md) for more details.
-->
7 changes: 7 additions & 0 deletions .github/SECURITY.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
# Security Policy

We aim to make YOLOv5 🚀 as secure as possible! If you find potential vulnerabilities or have any concerns please let us know so we can investigate and take corrective action if needed.

### Reporting a Vulnerability

To report vulnerabilities please email us at hello@ultralytics.com or visit https://ultralytics.com/contact. Thank you!
31 changes: 21 additions & 10 deletions .github/dependabot.yml
Original file line number Diff line number Diff line change
@@ -1,12 +1,23 @@
version: 2
updates:
- package-ecosystem: pip
directory: "/"
schedule:
interval: weekly
time: "04:00"
open-pull-requests-limit: 10
reviewers:
- glenn-jocher
labels:
- dependencies
- package-ecosystem: pip
directory: "/"
schedule:
interval: weekly
time: "04:00"
open-pull-requests-limit: 10
reviewers:
- glenn-jocher
labels:
- dependencies

- package-ecosystem: github-actions
directory: "/"
schedule:
interval: weekly
time: "04:00"
open-pull-requests-limit: 5
reviewers:
- glenn-jocher
labels:
- dependencies
Loading