0810 개발일지

이나겸·2022년 8월 10일
0

1. 개발기록

  • azure에서 데이터 라벨링을 하면 coco 형식의 정규화된 값으로 나와서
    그 값을 다시 원 이미지의 가로, 세로 크기로 곱하고
    그걸 yolo 형식으로 다시 정규화 해서 txt 파일로 추출하도록 했다.
  • 총 2000장 중에서 skip으로 넘긴 사진을 제외하고 1000장만 라벨링 하기로 함.
    이미지 이름과 라벨링한 파일 이름을 매칭해서 맞으면 shutil.move로 이동시킴.
    딱 1000장이 이동되는 것을 확인했음.
import os
import glob
import shutil

labels_path = glob.glob(os.path.join("./refine_plate_split/L1000_labels", "*.txt"))
images_path = glob.glob(os.path.join("./refine_plate_split/L1000", "*.jpg"))

os.makedirs("./refine_plate_split/L1000_images", exist_ok=True)
for label in labels_path:
    label_name = label.split('\\')[-1].split('.')[0]
    for image in images_path :
        if label_name in image:
            shutil.move(image, "./refine_plate_split/L1000_images/")
        else:
            continue

0개의 댓글