MMOCR 學習紀錄

參考 https://mmocr.readthedocs.io/en/dev-1.x/

參考 https://github.com/open-mmlab/mmocr

$ docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:21.05-py3

$ docker run --gpus all -it --name MMOCR nvcr.io/nvidia/pytorch:21.05-py3

$ docker start MMOCR

$ docker attach MMOCR

# <ctrl+p><ctrl+q>

$ docker attach MMOCR

$ docker stop MMOCR

$ docker rm MMOCR

$ docker run --gpus all -it --name MMOCR --shm-size=8G \

  -v /mnt/Data/MMOCR/mmocr:/workspace/mmocr \

  -v /mnt/QNAP_A/ImageData/ICDAR:/mmocr/data \

  nvcr.io/nvidia/pytorch:21.05-py3

# pip install -U openmim

#### 出現下列錯誤

####ERROR: After October 2020 you may experience errors when installing or updating packages. This is because pip will change the way that it resolves dependency conflicts.

####We recommend you use --use-feature=2020-resolver to test your packages with the new resolver before it becomes the default.

#### 不用擔心

確認 https://mmocr.readthedocs.io/en/dev-1.x/get_started/install.html 底部的版本資訊

安裝正確版本套件

# mim list

# mim install mmengine==

# mim install mmengine

# mim install mmcv==2.0.1

#### 出現下列錯誤

####    cv.gapi.wip.GStreamerPipeline = cv.gapi_wip_gst_GStreamerPipeline

####AttributeError: partially initialized module 'cv2' has no attribute 'gapi_wip_gst_GStreamerPipeline' (most likely due to a circular import)

#### 解決

#### 因為目前 opencv-python 版本為 4.10.0.84, 降版本

# pip install opencv-python==4.5.1.48

# mim install mmcv==2.0.1

# mim install mmdet==3.1.0

# cd /workspace/mmocr/

# pip install -v -e .

# pip install opencv-python-headless==4.5.1.48

# pip install -r requirements/albu.txt

# pip install -r requirements.txt

# python tools/infer.py demo/images --det DBNet --rec CRNN --print-result \

  --save_pred --save_vis --out-dir='results/' --batch-size=2