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

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

Prize : 1,000 만원
2,093명 마감

Practice

Overview Data Code (share) Talk Leaderboard

Submission

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

stevekim

2023.04.15 09:45 2,725 Views language

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

Code

Comments 1개

로그인이 필요합니다

comment

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에 활용해도 괜찮은거 같아요.

📣 댓글 작성 창의 위치가 댓글 리스트 상단으로 이동하였습니다!

List

[utility notebook]train.csv, label_info.csv 생성 노트북

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

현재 글

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

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

Likes 14