From a3618b86efddaddd2ee9f12bc623be817295b464 Mon Sep 17 00:00:00 2001
From: TannedCung <dx.tan@samsung.com>
Date: Thu, 27 Jun 2024 17:21:47 +0700
Subject: [PATCH] Fix: Deskew and text detection process

---
 cope2n-ai-fi/Dockerfile                   | 11 +++--------
 cope2n-ai-fi/configs/sdsap_sbt/configs.py |  1 +
 2 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/cope2n-ai-fi/Dockerfile b/cope2n-ai-fi/Dockerfile
index c86986d..5040616 100755
--- a/cope2n-ai-fi/Dockerfile
+++ b/cope2n-ai-fi/Dockerfile
@@ -15,18 +15,13 @@ RUN pip install mmcv==1.6.0 -f https://download.openmmlab.com/mmcv/dst/cu116/tor
 RUN ln -s /opt/conda/lib/python3.10/site-packages/torch/lib/libcudnn.so.8 /usr/lib/libcudnn.so && \
     ln -s /opt/conda/lib/libcublas.so /usr/lib/libcublas.so
 
-# RUN python -m pip install paddlepaddle-gpu==2.4.2.post116 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html --no-cache-dir
 # RUN pip install fastdeploy-gpu-python==1.0.7 -f https://www.paddlepaddle.org.cn/whl/fastdeploy.html --no-cache-dir
-RUN wget https://bj.bcebos.com/fastdeploy/release/wheels/fastdeploy_python-1.0.7-cp310-cp310-manylinux1_x86_64.whl \
-    && pip install fastdeploy_python-1.0.7-cp310-cp310-manylinux1_x86_64.whl --no-cache-dir \
-    && rm fastdeploy_python-1.0.7-cp310-cp310-manylinux1_x86_64.whl
+RUN wget https://bj.bcebos.com/fastdeploy/release/wheels/fastdeploy_gpu_python-1.0.7-cp310-cp310-manylinux1_x86_64.whl \
+    && pip install fastdeploy_gpu_python-1.0.7-cp310-cp310-manylinux1_x86_64.whl --no-cache-dir \
+    && rm fastdeploy_gpu_python-1.0.7-cp310-cp310-manylinux1_x86_64.whl
 
 RUN python -m pip install 'git+https://github.com/facebookresearch/detectron2.git@a59f05630a8f205756064244bf5beb8661f96180' --no-cache-dir
 
-RUN wget https://paddle-wheel.bj.bcebos.com/2.4.2/linux/linux-gpu-cuda11.6-cudnn8.4.0-mkl-gcc8.2-avx/paddlepaddle_gpu-2.4.2.post116-cp310-cp310-linux_x86_64.whl \
-    && pip install paddlepaddle_gpu-2.4.2.post116-cp310-cp310-linux_x86_64.whl --no-cache-dir \
-    && rm paddlepaddle_gpu-2.4.2.post116-cp310-cp310-linux_x86_64.whl
-
 # Install SDSV packages
 COPY . /workspace/cope2n-ai-fi
 
diff --git a/cope2n-ai-fi/configs/sdsap_sbt/configs.py b/cope2n-ai-fi/configs/sdsap_sbt/configs.py
index 855e26b..c312022 100755
--- a/cope2n-ai-fi/configs/sdsap_sbt/configs.py
+++ b/cope2n-ai-fi/configs/sdsap_sbt/configs.py
@@ -3,6 +3,7 @@ ocr_engine = {
     "detector": {
         "version": "/workspace/cope2n-ai-fi/weights/models/ocr_engine/sdsvtd/27Dec_washing420_79_epoch_100_params.pth",
         "rotator_version": "/workspace/cope2n-ai-fi/weights/models/ocr_engine/sdsvtd/best_bbox_mAP_epoch_30_lite.pth",
+        "auto_rotate": False,
         "device": device
     },
     "recognizer": {