도배 하자 유형 분류 AI 경진대회

알고리즘 | 비전 | 분류 | MLOps | Weighted F1 Score

  • moneyIcon 상금 : 1,000 만원
  • 2,062명 마감

 

별건 아니지만 데이터 모으고 df로 정리코드 (+ 한글 이미지 읽기/쓰기)

2023.04.15 09:45 2,135 조회 language

데이터를 한 디렉토리에 모으고 한번에 df로 정리하고자 만들었습니다
이렇게 하면 다른 task에도 적용하기 쉽고 Labelencoding하기도 쉽습니다
한글이미지 적용도 가능하므로 이름 안바꾸셔도 됩니다

코드
로그인이 필요합니다
0 / 1000
KJC
2023.04.16 00:17

def search(dirname):
    dirs = []
    filenames = os.listdir(dirname)
    for filename in filenames:
        full_filename = os.path.join(dirname, filename)
        dirs.append(os.path.basename(full_filename))
    return dirs
로 folder들을 list로 만들고, class들을 0~18 까지 할당한다음 inference에 활용해도 괜찮은거 같아요.