Update pixelconversion.py

pixelconversion.py

@@ -23,26 +23,31 @@ def openDrawPDF(path):
   doc = fitz.open(path)
   out = fitz.open()  # output PDF
   page = doc[0]
+  print(page.rotation)
   pix = page.get_pixmap()  # render page to an image
   pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)
   img=np.array(pl)
 
   img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
-  ratio =  pix.width / img.shape[1]  
-
-  img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-  imgBW=cv2.threshold(img, 250, 255, cv2.THRESH_BINARY_INV)[1]
+  ratio =  pix.width / img.shape[1]
 
+  imgGry = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+  imgBW=cv2.threshold(imgGry, 250, 255, cv2.THRESH_BINARY_INV)[1]
   contours, hierarchy = cv2.findContours(imgBW, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
-  img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
 
-  contour=contours[2]
-  perimeter = cv2.arcLength(contour, True)
-  approx = cv2.approxPolyDP(contour, 0.01* perimeter, True)
+  newcontours = contours[2:]
+  areas = [cv2.contourArea(c) for c in newcontours]
+  max_index2 = np.argmax(areas)
+  contour=newcontours[max_index2]
+  
+  area=cv2.contourArea(contour)
+  if area >500:
+    perimeter = cv2.arcLength(contour, True)
+    approx = cv2.approxPolyDP(contour, 0.01* perimeter, True)
   shape=[]
   for point in approx:
       x1, y1 = point[0]
-      
+
       shape.append([(int(x1*ratio)),(int(y1*ratio))])
       cv2.circle (img, (x1, y1), 5, 255, 5)