main.py

import cv2
import numpy as np
from svm import SingleViewM
import os

# lines = {
#     "x": [[(205, 280), (315, 220)], [(205, 205), (318, 150)], [(87, 121), (197, 74)]],
#     "y": [[(205, 280), (90, 190)], [(205, 205), (87, 121)], [(318, 150), (197, 74)]],
#     "z": [[(205, 280), (205, 205)], [(90, 190), (86, 121)], [(315, 220), (320, 150)]]
# }
# refpoints = {
#     'x': np.array([315, 220, 1]).reshape(-1, 1),
#     'y': np.array([90, 190, 1]).reshape(-1, 1),
#     'z': np.array([205, 205, 1]).reshape(-1, 1),
#     'o': np.array([205, 280, 1])
# }

if __name__ == '__main__':
    lines = {
        "x": [[(1700, 3666), (3289, 2911)], [(1740, 3366), (3355, 2670)], [(259, 1921), (1553, 1784)]],
        "y": [[(1700, 3666), (262, 2079)], [(1740, 3366), (259, 1921)], [(3355, 2670), (1553, 1784)]],
        "z": [[(1700, 3666), (1740, 3366)], [(3289, 2911), (3355, 2670)], [(262, 2079), (259, 1921)]]
    }
    refpoints = {
        'x': np.array([3289, 2911, 1]).reshape(-1, 1),
        'y': np.array([262, 2079, 1]).reshape(-1, 1),
        'z': np.array([1740, 3366, 1]).reshape(-1, 1),
        'o': np.array([1700, 3666, 1])
    }
    img = cv2.imread('img/onepiece.jpg')
    singlevm = SingleViewM(img, endpoints=lines, refpoints=refpoints)

    proj_matrix = singlevm.findProjectionMatrix()
    H_xy = proj_matrix[:, [0, 1, 3]].copy()
    H_yz = -proj_matrix[:, [1, 2, 3]].copy()
    H_xz = -proj_matrix[:, [0, 2, 3]].copy()

    if not os.path.isdir('textures'):
        os.mkdir('textures')

    frame_xy = cv2.warpPerspective(img, H_xy, img.shape[:2], flags=cv2.WARP_INVERSE_MAP)
    cv2.imwrite('textures/xy.jpg', frame_xy)

    frame_xz = cv2.warpPerspective(img, H_xz, img.shape[:2], flags=cv2.WARP_INVERSE_MAP)
    cv2.imwrite('textures/xz.jpg', frame_xz)

    frame_yz = cv2.warpPerspective(img, H_yz, img.shape[:2], flags=cv2.WARP_INVERSE_MAP)
    cv2.imwrite('textures/yz.jpg', frame_yz)