Glenn Jocher
ecce7c86c7
update comments
2021-12-19 18:04:52 +01:00
Glenn Jocher
d5fc52132a
list fix
2021-12-19 18:04:04 +01:00
Glenn Jocher
6650720fbf
Update
2021-12-19 17:55:16 +01:00
Glenn Jocher
1ec653b06b
Update
2021-12-19 17:14:20 +01:00
Glenn Jocher
5e09d0424c
Update
2021-12-19 17:08:15 +01:00
Glenn Jocher
b5954cc30a
Update
2021-12-19 17:06:23 +01:00
Glenn Jocher
a3f21ad154
Update
2021-12-19 17:05:21 +01:00
Glenn Jocher
49ff485e5e
Update
2021-12-19 16:39:58 +01:00
Glenn Jocher
73db0effec
Update
2021-12-19 16:35:48 +01:00
Glenn Jocher
63f319ac3d
Update
2021-12-19 16:34:56 +01:00
Glenn Jocher
664468eecf
Update
2021-12-19 16:14:44 +01:00
Glenn Jocher
8342c5a158
Merge master
2021-12-19 16:08:26 +01:00
Glenn Jocher
3a2417430a
Update show
2021-12-19 16:04:43 +01:00
Glenn Jocher
26f0415287
Add dataset source citations ( #6032 )
2021-12-19 15:19:04 +01:00
Glenn Jocher
abbdd4802e
train -> val comment fix ( #6024 )
2021-12-17 16:43:00 +01:00
Felix You
361705d9be
Multi-layer capable --freeze argument ( #6019 )
...
* support specfiy multiple frozen layers
* fix bug
* Cleanup Freeze section
* Cleanup argument
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-17 15:42:26 +01:00
Glenn Jocher
407a905747
Check TensorRT>=8.0.0 version ( #6021 )
...
* Check TensorRT>=8.0.0 version
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-12-17 14:59:46 +01:00
Glenn Jocher
c1249a47c7
*.torchscript inference self.jit fix ( #6007 )
2021-12-16 14:10:54 +01:00
Glenn Jocher
628817dfae
Recommend jar xf file.zip for zips ( #5993 )
2021-12-15 17:19:19 +01:00
Glenn Jocher
da9a1b719b
Allow --weights URL ( #5991 )
2021-12-15 15:27:08 +01:00
Mrinal Jain
b7d18f3636
Increase ar_thr from 20 to 100 for better detection on slender (high aspect ratio) objects ( #5556 )
...
* Making `ar_thr` available as a hyperparameter
* Disabling ar_thr as hyperparameter and computing from the dataset instead
* Fixing bug in ar_thr computation
* Fix `ar_thr` to 100
2021-12-15 14:42:23 +01:00
Glenn Jocher
c9a46a60e0
Update callbacks.py with __init__() ( #5979 )
...
Add __init__() function.
2021-12-14 15:47:49 +01:00
Glenn Jocher
d699c21c75
YOLOv5s6 params and FLOPs fix ( #5977 )
2021-12-14 11:24:39 +01:00
jinmc
2d0c6afbfe
make parameter ignore epochs ( #5972 )
...
* make parameter ignore epochs
ignore epochs functionality add to prevent spikes at the beginning when fitness spikes and decreases after.
Discussed at https://github.com/ultralytics/yolov5/issues/5971
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-14 11:18:34 +01:00
Glenn Jocher
e8ef8fb1ca
pretrained=False fix (#5966 )
...
* `pretriained=False` fix
Fix for https://github.com/ultralytics/yolov5/issues/5964
* CI speed improvement
2021-12-13 13:32:27 +01:00
Glenn Jocher
a6037838a3
Merge master
2021-12-12 23:52:55 +01:00
Diego Montes
19c56e60b1
Fix imgsz bug ( #5948 )
...
* fix imgsz bug
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-12 23:39:14 +01:00
Yono Mittlefehldt
8f354362cd
Fix Detections class tolist() method ( #5945 )
...
* Fix tolist() to add the file for each Detection
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Fix PEP8 requirement for 2 spaces before an inline comment
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-11 18:40:37 +01:00
Glenn Jocher
ea3d131010
Update default --workers 8
2021-12-11 14:34:12 +01:00
Glenn Jocher
6c2ac617d9
Merge master
2021-12-11 14:32:39 +01:00
Glenn Jocher
8f875d93a2
Refactor NUM_THREADS ( #5954 )
2021-12-11 14:32:18 +01:00
Glenn Jocher
ecd0f6d9b9
Add mnist-fashion
2021-12-11 13:36:02 +01:00
Glenn Jocher
d88ee1e629
Merge master
2021-12-11 13:08:42 +01:00
Glenn Jocher
14af287e6c
Add FILE ROOT globals
2021-12-11 13:00:42 +01:00
Diego Montes
2c6317547a
Add nms and agnostic nms to export.py ( #5938 )
...
* add nms and agnostic nms to export.py
* fix agnostic implies nms
* reorder args to group TF args
* PEP8 120 char
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-10 18:24:32 +01:00
iumyx2612
a42af30d8a
Update strip_optimizer() ( #5949 )
...
Replace 'training_result' with 'best_fitness' in strip_optimizer() to match key with ckpt from train.py
2021-12-10 18:06:27 +01:00
Pascal Maillard
922fbd8209
fix .gitignore not tracking existing folders ( #5946 )
...
* fix .gitignore not tracking existing folders
fix .gitignore so that the files that are in the repository are actually being tracked.
Everything in the data/ folder is ignored, which also means the subdirectories are ignored. Fix so that the subdirectories and their contents are still tracked.
* Remove data/trainings
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-10 17:32:09 +01:00
Felix You
c45f9f678d
Make select_device() robust to batch_size=-1 ( #5940 )
...
* Find out a bug. When set batch_size = -1 to use the autobatch.
reproduce:
* Fix type conflict
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-10 15:27:20 +01:00
Diego Montes
4fb6dd4b26
Fix ONNX opset inconsistency with parseargs and run args ( #5937 )
2021-12-09 23:10:16 +01:00
Glenn Jocher
5bdb28ed10
Default PyTorch Hub to autocast(False) ( #5926 )
2021-12-08 23:15:14 +01:00
Glenn Jocher
3e0008d53d
Add --adam to Usage: example
2021-12-08 17:48:05 +01:00
Glenn Jocher
2b40bd928a
Merge master
2021-12-08 17:32:20 +01:00
Glenn Jocher
c77a5a84e3
Absolute '/content/sample_data' ( #5922 )
2021-12-08 17:31:31 +01:00
Glenn Jocher
3feecfaeb7
Improve stability by disabling AMP
2021-12-08 17:19:58 +01:00
Glenn Jocher
a3d5f1d3e3
Revert "Update plot_lr_scheduler() ( #5864 )" ( #5920 )
...
This reverts commit 360eec69101ad9ffada78326b715b724b9b7eb0d.
2021-12-08 16:46:24 +01:00
Glenn Jocher
1207356411
Implement scaled loss backward
2021-12-08 16:33:16 +01:00
Glenn Jocher
d82766dd09
Merge master
2021-12-08 16:18:42 +01:00
Glenn Jocher
91e13b24f4
Smart DetectMultiBackend() unwrapping
2021-12-08 16:15:31 +01:00
Glenn Jocher
7d56d45124
Add hardware checks to notebook_init() ( #5919 )
...
* Update notebook
* Update notebook
* update string
* update string
* Updates
* Updates
* Updates
* check both ipython and psutil
* remove sample_data if is_colab
* cleanup
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-12-08 14:57:03 +01:00
Glenn Jocher
581dc301a7
Add ONNX inference providers ( #5918 )
...
* Add ONNX inference providers
Fix for https://github.com/ultralytics/yolov5/issues/5916
* Update common.py
2021-12-08 13:37:33 +01:00
