대학 대항전 : 퍼즐 이미지 AI 경진대회

알고리즘 | 월간 데이콘 | 대학 대항전 | 비전 | 자기지도학습 | 기술 혁신 | 분류

 

정렬된 이미지와 증강된 이미지 출력 및 저장하기 (Colab)

2024.01.02 21:52 3,708 조회 language

기존 train 이미지를 순서에 맞게 정렬하고 다른 순서로 배열하는 코드입니다
Colab 기준으로 저장하실 경우 ram 부족으로 다운될 수 있으니 코드를 수정하여 데이터를 나누어 저장하는 것을 권장합니다

코드
로그인이 필요합니다
0 / 1000
설빙더아이스
2024.01.03 19:00

좋은 코드 공유 감사합니다!

응애맨
2024.01.03 23:09

와 증강까지.. 

윈도우
2024.01.04 00:31

좋은 코드 감사합니다.

로컬에서 사용 중 메모리 누수 현상이 있어서 함수 99번째 줄 쯤 코드를
        if count > show_num:
            plt.cla()
            plt.clf()
            plt.close('all')
            pass
        else:
            print(train_path)
            plt.show()
            print()
            plt.cla()
            plt.clf()
            plt.close('all')
이렇게 수정해서 해결했습니다.