Glenn Jocher
0e2f2cbb51
Update LoadStreams init fallbacks ( #3295 )
vor 3 Jahren
Charles Frye
19100ba007
Improves docs and handling of entities and resuming by WandbLogger ( #3264 )
* adds latest tag to match wandb defaults
* adds entity handling, 'last' tag
* fixes bug causing finished runs to resume
* removes redundant "last" tag for wandb artifact
vor 3 Jahren
Glenn Jocher
683cefead4
YouTube stream ending fix ( #3277 )
* YouTube stream ending fix
Properly terminates YouTube streams on video end. Should resolve issues #2769 and #3220 .
* Update datasets.py
vor 3 Jahren
Glenn Jocher
f3402353fb
check `batch_size % utilized_device_count` ( #3276 )
Bug fix to check batch_size divisibility of utilized CUDA device count vs total system CUDA device count.
vor 3 Jahren
yeric1789
7b0eb952ca
`plot_one_box()` default `color=(128, 128, 128)` ( #3240 )
* Color can be none by default
* `plot_one_box()` default `color=(128, 128, 128)`
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
vor 3 Jahren
Adrian Holovaty
3f74cd9ed1
Parameterize max_det + inference default at 1000 ( #3215 )
* Added max_det parameters in various places
* 120 character line
* PEP8
* 120 character line
* Update inference default to 1000 instances
* Update inference default to 1000 instances
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
vor 3 Jahren
Glenn Jocher
13a1c72699
Update datasets.py ( #3216 )
vor 3 Jahren
Glenn Jocher
ffb47ffbeb
Silent Patches Bug Fix ( #3214 )
vor 3 Jahren
Glenn Jocher
36b0a9e7d0
Add CAP_PROP_FRAME_COUNT for YouTube sources ( #3193 )
vor 3 Jahren
Glenn Jocher
ae53f50ae7
Plot labels histogram colors ( #3192 )
vor 3 Jahren
Glenn Jocher
c1c7eb023f
Update JSON response ( #3139 )
vor 3 Jahren
Glenn Jocher
52c05707d6
Update Colors() ( #3046 )
* Update Colors()
* update colors
* update colors
vor 3 Jahren
Glenn Jocher
abfcf9eb79
Replace print() with logging.info() in trainloader ( #3103 )
Might indirectly help #3095 by providing better visibility on source of corruption.
vor 3 Jahren
Glenn Jocher
57b0d3a605
Add check_python() ( #3088 )
* Add check_python()
Checks python version against minimum version of 3.7.0.
* remove packaging dependency
* refactor import
vor 3 Jahren
Ayush Chaurasia
d2a17289c9
Explicitly convert artifact path to posix_path ( #3067 )
* Explicitly convert artifact path to posix_path
* Remove redudant str() casting
vor 3 Jahren
Ayush Chaurasia
b18ca31ff5
use check_file for --data ( #3035 )
vor 3 Jahren
Glenn Jocher
8cab907f60
download() ThreadPool update ( #3027 )
* download() ThreadPool update
* update train image count
* cid + 1
vor 3 Jahren
Glenn Jocher
5189b3addb
Objects365 Update
vor 3 Jahren
Glenn Jocher
41cc7caee6
Add NMS threshold checks ( #3020 )
* Add NMS threshold checks
* fix
vor 3 Jahren
Glenn Jocher
fcd57028d2
Add is_colab() function ( #3018 )
vor 3 Jahren
Glenn Jocher
31ee54c28c
Hubconf.py bug fix ( #3007 )
vor 3 Jahren
Glenn Jocher
37eaffec7d
Curl update ( #3004 )
* Curl update
* Curl update
vor 3 Jahren
Glenn Jocher
4d7bca7648
Add unzip flag to download() ( #3002 )
vor 3 Jahren
Glenn Jocher
54652fe3ff
Objects365 update
vor 3 Jahren
Ayush Chaurasia
801b469878
Improve performance of dataset Logger ( #2943 )
* Improve performance of Dataset Logger
* Fix scaling bug
vor 3 Jahren
Glenn Jocher
955eea8b96
Make cache saving optional ( #2977 )
vor 3 Jahren
Glenn Jocher
a833ee2a46
Update check_requirements() exclude list ( #2974 )
vor 3 Jahren
Glenn Jocher
2c7c075fb1
Update restapi.py ( #2967 )
vor 3 Jahren
Glenn Jocher
57812df68c
New Colors() class ( #2963 )
vor 3 Jahren
Glenn Jocher
1849916726
Check_requirements() enclosing apostrophe bug fix ( #2929 )
This fixes a bug where the '>' symbol in python package requirements was not running correctly with subprocess.check_output() commands.
vor 3 Jahren
Glenn Jocher
9c7bb5a52c
ACON Activation batch-size 1 bug patch ( #2901 )
* ACON Activation batch-size 1 bug path
This is not a great solution to https://github.com/nmaac/acon/issues/4 but it's all I could think of at the moment.
WARNING: YOLOv5 models with MetaAconC() activations are incapable of running inference at batch-size 1 properly due to a known bug in https://github.com/nmaac/acon/issues/4 with no known solution.
* Update activations.py
* Update activations.py
* Update activations.py
* Update activations.py
vor 3 Jahren
NanoCode012
c0d3f80544
Add verbose option to pytorch hub models ( #2926 )
* Add verbose and update print to logging
* Fix positonal param
* Revert auto formatting changes
* Update hubconf.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
vor 3 Jahren
Glenn Jocher
45632b2704
Update download() for tar.gz files ( #2919 )
* Update download() for tar.gz files
* Update general.py
vor 3 Jahren
Glenn Jocher
1b1ab4cca2
Add file_size() function ( #2911 )
* Add file_size() function
* Update export.py
vor 3 Jahren
Glenn Jocher
14d2d2d75f
Update google_utils.py ( #2900 )
vor 3 Jahren
Glenn Jocher
264d860f8d
ACON activation function ( #2893 )
* ACON Activation Function
## 🚀 Feature
There is a new activation function [ACON (CVPR 2021)](https://arxiv.org/pdf/2009.04759.pdf ) that unifies ReLU and Swish.
ACON is simple but very effective, code is here: https://github.com/nmaac/acon/blob/main/acon.py#L19
![image](https://user-images.githubusercontent.com/5032208/115676962-a38dfe80-a382-11eb-9883-61fa3216e3e6.png )
The improvements are very significant:
![image](https://user-images.githubusercontent.com/5032208/115680180-eac9be80-a385-11eb-9c7a-8643db552c69.png )
## Alternatives
It also has an enhanced version meta-ACON that uses a small network to learn beta explicitly, which may influence the speed a bit.
## Additional context
[Code](https://github.com/nmaac/acon ) and [paper](https://arxiv.org/pdf/2009.04759.pdf ).
* Update activations.py
vor 3 Jahren
r-blmnr
0395e39fd0
Uppercase model filenames enabled ( #2890 )
vor 3 Jahren
Glenn Jocher
78fd077657
VisDrone2019-DET Dataset Auto-Download ( #2882 )
* VisDrone Dataset Auto-Download
* add visdrone.yaml
* cleanup
* add VisDrone2019-DET-test-dev
* cleanup VOC
vor 3 Jahren
Michael Heilig
d48a34dca7
bug fix: switched rows and cols for correct detections in confusion matrix ( #2883 )
vor 3 Jahren
JoshSong
5f7d39fede
Cleanup load_image() ( #2871 )
* don't resize up in load_image if augmenting
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
vor 3 Jahren
Glenn Jocher
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
* Implement yaml.safe_load()
* yaml.safe_dump()
vor 3 Jahren
Burhan
c949fc86d1
Detection cropping+saving feature addition for detect.py and PyTorch Hub ( #2827 )
* Update detect.py
* Update detect.py
* Update greetings.yml
* Update cropping
* cleanup
* Update increment_path()
* Update common.py
* Update detect.py
* Update detect.py
* Update detect.py
* Update common.py
* cleanup
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
vor 3 Jahren
Glenn Jocher
c5c647e281
Update increment_path() to handle file paths ( #2867 )
vor 3 Jahren
Glenn Jocher
c15e25c40f
PyTorch Hub cv2 .save() .show() bug fix ( #2831 )
* PyTorch Hub cv2 .save() .show() bug fix
cv2.rectangle() was failing on non-contiguous np array inputs. This checks for contiguous arrays and applies is necessary:
```python
imgs[i] = im if im.data.contiguous else np.ascontiguousarray(im) # update
```
* Update plots.py
```python
assert im.data.contiguous, 'Image not contiguous. Apply np.ascontiguousarray(im) to plot_on_box() input image.'
```
* Update hubconf.py
Expand CI tests to OpenCV image.
vor 3 Jahren
Glenn Jocher
aff03be35a
YouTube Bug Fix ( #2818 )
Fix for #2810
```shell
python detect.py --source 0
```
introduced by YouTube Livestream Detection PR #2752
vor 3 Jahren
Glenn Jocher
1f3e482bce
ONNX Simplifier ( #2815 )
* ONNX Simplifier
Add ONNX Simplifier to ONNX export pipeline in export.py. Will auto-install onnx-simplifier if onnx is installed but onnx-simplifier is not.
* Update general.py
vor 3 Jahren
Robin
1479737064
Flask REST API Example ( #2732 )
* add files
* Update README.md
* Update README.md
* Update restapi.py
pretrained=True and model.eval() are used by default when loading a model now, so no need to call them manually.
* PEP8 reformat
* PEP8 reformat
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
vor 3 Jahren
Glenn Jocher
0f395b3e3b
YOLOv5 v5.0 Release patch 1 ( #2764 )
* torch.jit.trace(model, img, strict=False)
* Update check_file()
* Update hubconf.py
* Update README.md
vor 3 Jahren
Glenn Jocher
f5b8f7d54c
YOLOv5 v5.0 Release ( #2762 )
vor 3 Jahren
Ben Milanko
e2b7bc0b32
YouTube Livestream Detection ( #2752 )
* Youtube livestream detection
* dependancy update to auto install pafy
* Remove print
* include youtube_dl in deps
* PEP8 reformat
* youtube url check fix
* reduce lines
* add comment
* update check_requirements
* stream framerate fix
* Update README.md
* cleanup
* PEP8
* remove cap.retrieve() failure code
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
vor 3 Jahren