카메라 이미지 품질 향상 AI 경진대회

알고리즘 | CV | 이미지 변환 | 제어 | PSNR

  • moneyIcon 상금 : 총 1,000만원
  • 1,286명 마감

 

Split train, validation set

2021.07.01 20:25 4,820 조회 language

Camera_quality_improvement
.. └ * augmentation.ipynb *
.. └ models
              ...
.. └ dataset
....      ├ train_input_img
....      ├ train_label_img
....      ├ test_input_img
....      ├ sample_submission
....      ├ train.csv
....      └ test.csv

디렉토리 구조가 위와 같습니다. 아래의 함수는 전처리가 완료된 이미지가 포함된 디렉토리의 경로를 매개변수로 받아
train_size를 기준으로 랜덤하게 분할하여 train set과 validation set .txt 파일을 생성합니다.

코드