polar2cart.py

import numpy as np
import cv2


def polar2cart(img: np.ndarray):
    """
    Convert image (B-scan) from Polar coordinates to Cartesian coordinates
    """
    
    # Resize image to square first
    h, w = img.shape[:2]
    sz = min(h, w)
    r = sz // 2
    img = cv2.resize(img, (sz, sz))

    # warp - requires rotation
    img = cv2.rotate(img, cv2.ROTATE_90_COUNTERCLOCKWISE)
    img = cv2.warpPolar(img, img.shape[:2], (r, r), r, cv2.WARP_INVERSE_MAP)
    img = cv2.rotate(img, cv2.ROTATE_90_COUNTERCLOCKWISE)
    return img