인프의 코딩월드

deep learning/코딩

• 전처리) 파일 이름 변경하기 (data, label 이름 동시에 바꾸기)

  deep learning/코딩 2021. 12. 20. 20:19

  import os image_path = "my_image_data_path" label_path = "my_label_data_path" file_name= os.listdir(image_path) for i in range(len(file_name)): old_file_name = os.path.join(image_path,file_name[i]) label_name = file_name[i].replace(".jpg",".txt") # label의 형식이 txt일 경우 old_label_name = os.path.join(label_path,label_name) file_new_img_name = os.path.join(image_path, f"daejeon_jan_{i}.jpg") # 변경할 이미..

  Read More
• 전처리) train_list.와 val_list 파일 나누기

  deep learning/코딩 2021. 12. 2. 21:15

  그동안 Yolo와 같은 모델을 통해 학습하기 위해 정말 많은 전처리를 하게 되었다. 그 중엔 xml 파일을 yolo 스타일의 txt 파일로 변경하거나 json 파일을 똑같이 yolo txt 스타일로 변경하는 일을 자주 하게 되었다. 그동안 한번 코드를 짜둔 후 계속 복사해서 사용하다 보니 꽤나 불편하고 오랜만에 보면 기억도 잘 안나서 블로그에 정리해 두고 깃허브에 코드를 올려둘 계획이다. 아직 yolov5 이하의 모델을 사용해본적이 없어서 다른 모델은 모르겠지만 학습 중에 학습을 하고 그 것을 검증할 데이터 셋을 나눠야 한다. 아래 코드는 그 코드이다. from glob import glob path = "학습을 돌릴 path" #학습을 돌릴 path => train.txt와 val.txt를 저장할 pa..

  Read More