yolov5/models
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
..
hub Add TransformerLayer, TransformerBlock, C3TR modules (#2333) 2021-04-01 17:26:53 +02:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py PyTorch Hub cv2 .save() .show() bug fix (#2831) 2021-04-18 13:47:40 +02:00
experimental.py Update detections() self.t = tuple() (#2617) 2021-03-26 13:23:11 +01:00
export.py ONNX Simplifier (#2815) 2021-04-16 14:03:27 +02:00
yolo.py Add TransformerLayer, TransformerBlock, C3TR modules (#2333) 2021-04-01 17:26:53 +02:00
yolov5l.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5m.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5s.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5x.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00