BUG: cv2.resize(nl_map, dsize=(h, w))

[onlyairnopods]

https://github.com/AlexHex7/Non-local_pytorch/blame/ec15789af4533448169f3dc36abe542badd34221/nl_map_vis/nl_map_vis.py#L33

nl_map = cv2.resize(nl_map, dsize=(h, w))

Shouldn't it be dsize=(w, h) ?
I encountered this bug when my image size W ≠ H.