Merge pull request #9 from modelai/fzp/update_convertlabel_code

use imagesize to get img w and h instead of read img

Author: yzbx

det-yolov4-tmi/convert_label_ark2txt.py

@@ -1,6 +1,6 @@
 import os
+import imagesize
 
-import cv2
 
 
 def _annotation_path_for_image(image_path: str, annotations_dir: str) -> str:
@@ -30,10 +30,7 @@ def _convert_annotations(index_file_path: str, dst_annotations_dir: str) -> None
         # each_txtfile: annotation path
         each_imgpath, each_txtfile = each_img_anno_path.split()
 
-        img = cv2.imread(each_imgpath)
-        if img is None:
-            raise ValueError(f"can not read image: {each_imgpath}")
-        img_h, img_w, _ = img.shape
+        img_w, img_h = imagesize.get(each_imgpath)
 
         with open(each_txtfile, 'r') as f:
             txt_content = f.readlines()

det-yolov4-tmi/cuda101.dockerfile

@@ -15,7 +15,7 @@ RUN wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_o
 RUN rm /usr/bin/python3
 RUN ln -s /usr/bin/python3.7 /usr/bin/python3
 RUN python3 get-pip.py
-RUN pip3 install -i ${PIP_SOURCE} mxnet-cu101==1.5.1 numpy opencv-python pyyaml watchdog tensorboardX six scipy tqdm
+RUN pip3 install -i ${PIP_SOURCE} mxnet-cu101==1.5.1 numpy opencv-python pyyaml watchdog tensorboardX six scipy tqdm imagesize
 
 ENV DEBIAN_FRONTEND noninteractive
 RUN apt-get update && apt-get install -y libopencv-dev

det-yolov4-tmi/cuda112.dockerfile

@@ -15,7 +15,7 @@ RUN wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_o
 RUN rm /usr/bin/python3
 RUN ln -s /usr/bin/python3.7 /usr/bin/python3
 RUN python3 get-pip.py
-RUN pip3 install -i ${PIP_SOURCE} mxnet-cu112==1.9.1 numpy opencv-python pyyaml watchdog tensorboardX six scipy tqdm
+RUN pip3 install -i ${PIP_SOURCE} mxnet-cu112==1.9.1 numpy opencv-python pyyaml watchdog tensorboardX six scipy tqdm imagesize
 
 ENV DEBIAN_FRONTEND noninteractive
 RUN apt-get update && apt-get install -y libopencv-dev