yolov5/utils
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.
2021-04-18 13:47:40 +02:00
..
aws resume.py typo (#2603) 2021-03-25 20:55:20 +01:00
flask_rest_api Flask REST API Example (#2732) 2021-04-15 13:26:08 +02:00
google_app_engine adding the configuration to deploy yolov5 in in app engine (#964) 2020-09-22 11:48:44 -07:00
wandb_logging utils/wandb_logging PEP8 reformat (#2755) 2021-04-10 21:09:23 +02:00
__init__.py initial commit 2020-05-29 17:04:54 -07:00
activations.py nn.SiLU() citation correction (#1713) 2021-01-06 20:58:41 -08:00
autoanchor.py Scipy kmeans-robust autoanchor update (#2470) 2021-03-14 21:58:12 -07:00
datasets.py YouTube Bug Fix (#2818) 2021-04-16 17:58:28 +02:00
general.py ONNX Simplifier (#2815) 2021-04-16 14:03:27 +02:00
google_utils.py Update google_utils.py (#2017) 2021-01-22 15:08:42 -08:00
loss.py Add --label-smoothing eps argument to train.py (default 0.0) (#2344) 2021-03-29 18:45:46 +02:00
metrics.py Confusion matrix background axis swap (#2114) 2021-03-02 16:09:52 -08:00
plots.py PyTorch Hub cv2 .save() .show() bug fix (#2831) 2021-04-18 13:47:40 +02:00
torch_utils.py Speed profiling improvements (#2648) 2021-03-29 15:19:07 +02:00