Я хочу выполнить локализацию точки, используя CNN. У меня есть набор данных, содержащий изображения и координаты точек (цель) в отдельных XML файлах. Я преобразовал и сохранил изображения и цели в .npy файлах.
Я хочу выполнить увеличение данных, а затем разделить обучающий набор и проверочный набор.
img_aug = ImageDataGenerator(horizontal_flip = True)
tgt_aug = ImageDataGenerator(horizontal_flip = True)
img_gen = img_aug.flow_from_directory(
path_img,
class_mode = None
)
tgt_gen = tgt_aug.flow_from_directory(
path_tgt,
class_mode = None
)
Это правильный способ выполнить это увеличение данных? Если нет, как я могу это сделать?
Позже я хочу разделить эти дополненные данные в обучающем наборе и проверочном наборе, используя это:
from sklearn.model_selection import train_test_split
X_train, X_val, y_train, y_val = train_test_split(img_gen, tgt_gen, test_size = 0.2, random_state = 1)