From 7a575d6e83cf5959963faefb277defa543436dc4 Mon Sep 17 00:00:00 2001
From: algobytewise <algobytewise@gmail.com>
Date: Tue, 6 Apr 2021 16:42:05 +0530
Subject: [PATCH] fix matrices & image-path

---
 digital_image_processing/rotation/rotation.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/digital_image_processing/rotation/rotation.py b/digital_image_processing/rotation/rotation.py
index 7cf909179..8d3fbf895 100644
--- a/digital_image_processing/rotation/rotation.py
+++ b/digital_image_processing/rotation/rotation.py
@@ -1,10 +1,11 @@
 import cv2
 import numpy as np
 from matplotlib import pyplot as plt
+import os.path
 
 
 def get_rotation(
-    img: np.ndarray, pt1: np.float32, pt2: np.float32, rows: int, cols: int
+    img: np.ndarray, pt1: np.ndarray, pt2: np.ndarray, rows: int, cols: int
 ) -> np.ndarray:
     """
     Get image rotation
@@ -21,17 +22,17 @@ def get_rotation(
 
 if __name__ == "__main__":
     # read original image
-    image = cv2.imread("lena.jpg")
+    image = cv2.imread(os.path.dirname(__file__)  + "/../image_data/lena.jpg")
     # turn image in gray scale value
     gray_img = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     # get image shape
     img_rows, img_cols = gray_img.shape
 
     # set different points to rotate image
-    pts1 = np.float32([[50, 50], [200, 50], [50, 200]])
-    pts2 = np.float32([[10, 100], [200, 50], [100, 250]])
-    pts3 = np.float32([[50, 50], [150, 50], [120, 200]])
-    pts4 = np.float32([[10, 100], [80, 50], [180, 250]])
+    pts1 = np.array([[50, 50], [200, 50], [50, 200]], np.float32)
+    pts2 = np.array([[10, 100], [200, 50], [100, 250]], np.float32)
+    pts3 = np.array([[50, 50], [150, 50], [120, 200]], np.float32)
+    pts4 = np.array([[10, 100], [80, 50], [180, 250]], np.float32)
 
     # add all rotated images in a list
     images = [