[Keras] ImageDataGenerator class weight

ImageDataGenerator class weight

 

 

딥러닝시 이미지 데이터의 불균형 문제를 해결하기 위해 class에 따른 가중치를 다르게 부여할 수 있다.

 

from sklearn.utils import compute_class_weight
import numpy as np

train_classes = train_generator.classes

class_weights = compute_class_weight(
                                        class_weight = "balanced",
                                        classes = np.unique(train_classes),
                                        y = train_classes                                                    
                                    )
                                    
class_weights = dict(zip(np.unique(train_classes), class_weights))

model.fit_generator(..., class_weight=train_class_weights)

 

balancing an imbalanced dataset with keras image generator - Stack Overflow

balancing an imbalanced dataset with keras image generator