람쥐썬더

프로덕트에 신경쓴다고 반년가까이 놓아버려 레퍼런스 없이 하지 못해 복습 겸 정리 기준은 이미지 다중 분류모델이다. 데이터는 이미지와 이미지 정보를 담은 DataFrame 이며 데이터는 StratifiedKFold 를 이용해 Fold 되어있는 상태 Pandas DataFrame 예시) image_path label fold path1/path2/image1.jpg 1 1 Custom DataLoader from torch.utils.data import Dataset, DataLoader import cv2 import pandas as pd import numpy as np class CustomDataGenerator(Dataset): def __init__(self, data, mode, batch_..