phucsd commited on
Commit
4df8b27
·
verified ·
1 Parent(s): c81c228

fix(generator): add missing PIL ImageFont imports and install git in Dockerfile

Browse files
Dockerfile CHANGED
@@ -5,6 +5,7 @@ ENV DEBIAN_FRONTEND=noninteractive
5
 
6
  # Install system dependencies required for OpenCV, PaddlePaddle, and OpenMP
7
  RUN apt-get update && apt-get install -y --no-install-recommends \
 
8
  libgl1 \
9
  libglib2.0-0 \
10
  libgomp1 \
 
5
 
6
  # Install system dependencies required for OpenCV, PaddlePaddle, and OpenMP
7
  RUN apt-get update && apt-get install -y --no-install-recommends \
8
+ git \
9
  libgl1 \
10
  libglib2.0-0 \
11
  libgomp1 \
ocr-training/scripts/generate_data.py CHANGED
@@ -5,6 +5,7 @@ import cv2
5
  import numpy as np
6
  import json
7
  import re
 
8
  try:
9
  from fontTools.ttLib import TTFont
10
  except ImportError:
 
5
  import numpy as np
6
  import json
7
  import re
8
+ from PIL import Image, ImageFont, ImageDraw
9
  try:
10
  from fontTools.ttLib import TTFont
11
  except ImportError: