search

rigvedrs / YOLO-V8-CAM

Wanna know what your model sees? Here's a package for applying EigenCAM on the new YOLO V8 model
MIT License
130 stars 26 forks source link

Confused width and height #6

Closed kolibriks closed 1 year ago

kolibriks commented 1 year ago

yolo_cam/base_cam.py def get_target_width_height

code: width, height = np.shape(input_tensor)[0],np.shape(input_tensor)[1] problem: np.shape returns the height first, then the width

fix: height, width = np.shape(input_tensor)[0],np.shape(input_tensor)[1]

kolibriks commented 1 year ago

created pull request to fix this issue https://github.com/rigvedrs/YOLO-V8-CAM/pull/7#issue-1912152498