Skip to content
  • Glenn Jocher's avatar
    PyTorch Hub cv2 .save() .show() bug fix (#2831) · c15e25c4
    Glenn Jocher authored
    * 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.
    c15e25c4