3665c0f59b
test.py native --single-cls ( #2928 )
3 years ago
aa78069c58
Change default value of hide-conf argument to false ( #2925 )
3 years ago
eae28a93b0
Change default value of hide label argument to False ( #2923 )
3 years ago
de19d396e7
Update visdrone.yaml ( #2921 )
3 years ago
45632b2704
Update download() for tar.gz files ( #2919 )
* Update download() for tar.gz files
* Update general.py
3 years ago
1b1ab4cca2
Add file_size() function ( #2911 )
* Add file_size() function
* Update export.py
3 years ago
646386ff09
Update export.py for 2 dry runs ( #2910 )
* Update export.py for 2 dry runs
* Update export.py
3 years ago
a2a514dec8
Update export.py ( #2909 )
3 years ago
28db237639
Default optimize_for_mobile() on TorchScript models ( #2908 )
Per https://pytorch.org/tutorials/recipes/script_optimized.html this should improve performance on torchscript models (and maybe coreml models also since coremltools operates on a torchscript model input, though this still requires testing).
3 years ago
f662c18507
Add detect.py --hide-conf --hide-labels --line-thickness options ( #2658 )
* command line option for line thickness and hiding labels
* command line option for line thickness and hiding labels
* command line option for line thickness and hiding labels
* command line option for line thickness and hiding labels
* command line option for line thickness and hiding labels
* command line option for hiding confidence values
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
14d2d2d75f
Update google_utils.py ( #2900 )
3 years ago
cdb678f418
Update yolo.py ( #2899 )
3 years ago
b40dd99167
Explicit opt function arguments ( #2817 )
* more explicit function arguments
* fix typo in detect.py
* revert import order
* revert import order
* remove default value
3 years ago
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
The improvements are very significant:
## 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
3 years ago
0395e39fd0
Uppercase model filenames enabled ( #2890 )
3 years ago
78fd077657
VisDrone2019-DET Dataset Auto-Download ( #2882 )
* VisDrone Dataset Auto-Download
* add visdrone.yaml
* cleanup
* add VisDrone2019-DET-test-dev
* cleanup VOC
3 years ago
d48a34dca7
bug fix: switched rows and cols for correct detections in confusion matrix ( #2883 )
3 years ago
5f7d39fede
Cleanup load_image() ( #2871 )
* don't resize up in load_image if augmenting
* cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
* Implement yaml.safe_load()
* yaml.safe_dump()
3 years ago
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>
3 years ago
c5c647e281
Update increment_path() to handle file paths ( #2867 )
3 years ago
1df8c6c963
Fix ONNX dynamic axes export support with onnx simplifier, make onnx simplifier optional ( #2856 )
* Ensure dynamic export works succesfully, onnx simplifier optional
* Update export.py
* add dashes
Co-authored-by: Tim <tim.stokman@hal24k.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
238583b7d5
Update FUNDING.yml ( #2833 )
* Update FUNDING.yml
* move FUNDING.yml to ./github
3 years ago
803f51bcee
Create FUNDING.yml ( #2832 )
3 years ago
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.
3 years ago
aff03be35a
YouTube Bug Fix ( #2818 )
Fix for #2810
```shell
python detect.py --source 0
```
introduced by YouTube Livestream Detection PR #2752
3 years ago
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
3 years ago
e5d71223b8
Update README.md
3 years ago
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>
3 years ago
1487bc84ff
Update README.md
3 years ago
860ca98832
Created using Colaboratory
3 years ago
cac8a765c8
Created using Colaboratory
3 years ago
2eab46e2cf
Update tutorial.ipynb
3 years ago
6b718e9127
Created using Colaboratory
3 years ago
54d65160b7
Update tutorial.ipynb
3 years ago
0f395b3e3b
YOLOv5 v5.0 Release patch 1 ( #2764 )
* torch.jit.trace(model, img, strict=False)
* Update check_file()
* Update hubconf.py
* Update README.md
3 years ago
f5b8f7d54c
YOLOv5 v5.0 Release ( #2762 )
3 years ago
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>
3 years ago
9029759cb3
Created using Colaboratory
3 years ago
6dd1083bbb
Tensorboard model visualization bug fix ( #2758 )
This fix should allow for visualizing YOLOv5 model graphs correctly in Tensorboard by uncommenting line 335 in train.py:
```python
if tb_writer:
tb_writer.add_graph(torch.jit.trace(model, imgs, strict=False), []) # add model graph
```
The problem was that the detect() layer checks the input size to adapt the grid if required, and tracing does not seem to like this shape check (even if the shape is fine and no grid recomputation is required). The following will warn:
0cae7576a9/train.py (L335)https://user-images.githubusercontent.com/26833433/114286928-349bd600-9a63-11eb-941f-7139ee6cd602.png ">
3 years ago
0cae7576a9
utils/wandb_logging PEP8 reformat ( #2755 )
* wandb_logging PEP8 reformat
* Update wandb_utils.py
3 years ago
b5de52c4cd
torch.cuda.amp bug fix ( #2750 )
PR https://github.com/ultralytics/yolov5/pull/2725 introduced a very specific bug that only affects multi-GPU trainings. Apparently the cause was using the torch.cuda.amp decorator in the autoShape forward method. I've implemented amp more traditionally in this PR, and the bug is resolved.
3 years ago
fca5e2a48f
autocast enable=torch.cuda.is_available() ( #2748 )
3 years ago
c03d590320
Add Hub results.pandas() method ( #2725 )
* Add Hub results.pandas() method
New method converts results from torch tensors to pandas DataFrames with column names.
This PR may partially resolve issue https://github.com/ultralytics/yolov5/issues/2703
```python
results = model(imgs)
print(results.pandas().xyxy[0])
xmin ymin xmax ymax confidence class name
0 57.068970 391.770599 241.383545 905.797852 0.868964 0 person
1 667.661255 399.303589 810.000000 881.396667 0.851888 0 person
2 222.878387 414.774231 343.804474 857.825073 0.838376 0 person
3 4.205386 234.447678 803.739136 750.023376 0.658006 5 bus
4 0.000000 550.596008 76.681190 878.669922 0.450596 0 person
```
* Update comments
torch example input now shown resized to size=640 and also now a multiple of P6 stride 64 (see https://github.com/ultralytics/yolov5/issues/2722#issuecomment-814785930 )
* apply decorators
* PEP8
* Update common.py
* pd.options.display.max_columns = 10
* Update common.py
3 years ago
c8c8da6079
Update README with collapsable notes ( #2721 )
* Update README with collapsable notes.
* cleanup
* center table
3 years ago
3067429307
Add support for list-of-directory data format for wandb ( #2719 )
3 years ago
ec8979f1d2
Updated filename attributes for YOLOv5 Hub BytesIO ( #2718 )
Fix 2 for 'Model predict with forward will fail if PIL image does not have filename attribute' #2702
3 years ago
74276d5189
Updated filename attributes for YOLOv5 Hub results ( #2708 )
Proposed fix for 'Model predict with forward will fail if PIL image does not have filename attribute' #2702
3 years ago
9ccfa85249
pip install coremltools onnx ( #2690 )
Requested in https://github.com/ultralytics/yolov5/issues/2686
3 years ago
17300a4c7b
autoShape forward im = np.asarray(im) # to numpy ( #2689 )
Slight speedup.
3 years ago
Glenn Jocher
albinxavi
Maximilian Peters
fcakyon
r-blmnr
Michael Heilig
JoshSong
Burhan
Tim Stokman
Robin
Ben Milanko
Ayush Chaurasia
