Private 5위 (0.85487), Convnext_large + Data Augmentation

월간 데이콘 예술 작품 화가 분류 AI 경진대회

정신이드는가닝겐

2022.11.15 16:23 1,733 조회 language

안녕하세요, JangJH입니다.
좋은 대회 열어 주셔서 감사합니다.
이번 대회에서 사용한 코드 공유드립니다.
보완할 점이나 궁금한 부분 있으시면 댓글 남겨주세요.

코드

댓글 3개

로그인이 필요합니다

comment

0 / 1000

kdy0912

2022.11.26 10:12

안녕하세요! 먼저 코드 공유 감사드립니다. 올려주신 코드를 좀 활용해 보려고 하는데 5-fold run 부분을 실행하니
error: OpenCV(4.6.0) /io/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'
이런 에러가 뜨는데 혹시 해결 방법 아실까요?? 제가 아직 코드가 익숙하지 않아 해결이 잘 안되네요ㅜㅜ

정신이드는가닝겐

2022.11.26 11:07

넵 안녕하세요! data load부분에서 파일 경로 문제로 이미지를 못 불러와서 에러가 생기는 것 같습니다.
data folder를 저는 따로 지정해주어서 파일 경로를 바꿔주셔야 될 것 같습니다.

def get_data(df, infer=False):

    df['img_path'] = df['img_path'].apply(lambda x: x[:2] + 'data/' + x[2:])

    if infer:
        return df['img_path'].values
    return df['img_path'].values, df['artist'].values

# def get_data(df, infer=False):
#     if infer:
#         return df['img_path'].values
#     return df['img_path'].values, df['artist'].values

코드에서 위에 함수를 주석처리하시고 밑에 함수로 사용해보시면 될 것 같습니다.