Commit 902dcd2c authored by myhloli's avatar myhloli
Browse files

refactor(BatchAnalyze): comment out image rotation logic in doclayout_yolo

parent c20e9a1e
...@@ -44,16 +44,16 @@ class BatchAnalyze: ...@@ -44,16 +44,16 @@ class BatchAnalyze:
modified_images = [] modified_images = []
for image_index, image in enumerate(images): for image_index, image in enumerate(images):
pil_img = Image.fromarray(image) pil_img = Image.fromarray(image)
width, height = pil_img.size # width, height = pil_img.size
if height > width: # if height > width:
input_res = {'poly': [0, 0, width, 0, width, height, 0, height]} # input_res = {'poly': [0, 0, width, 0, width, height, 0, height]}
new_image, useful_list = crop_img( # new_image, useful_list = crop_img(
input_res, pil_img, crop_paste_x=width // 2, crop_paste_y=0 # input_res, pil_img, crop_paste_x=width // 2, crop_paste_y=0
) # )
layout_images.append(new_image) # layout_images.append(new_image)
modified_images.append([image_index, useful_list]) # modified_images.append([image_index, useful_list])
else: # else:
layout_images.append(pil_img) layout_images.append(pil_img)
images_layout_res += self.model.layout_model.batch_predict( images_layout_res += self.model.layout_model.batch_predict(
layout_images, self.batch_ratio * YOLO_LAYOUT_BASE_BATCH_SIZE layout_images, self.batch_ratio * YOLO_LAYOUT_BASE_BATCH_SIZE
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment