train_ds_cls = DataManager(
tfrecord_path=path_cls.get_train_ds_path(),
img_root=IMAGE_ROOT_PATH,
batch_size=LEARN_BATCH_SIZE,
net_cls=net_cls,
data_n=df.at[0, 'total_learn_data'],
)
test_ds_cls = DataManager(
tfrecord_path=path_cls.get_test_ds_path(),
img_root=IMAGE_ROOT_PATH,
batch_size=TEST_BATCH_SIZE,
net_cls=net_cls,
data_n=df.at[0, 'total_test_data'],
)
valid_ds_cls = DataManager(
tfrecord_path=path_cls.get_validation_ds_path(),
img_root=IMAGE_ROOT_PATH,
batch_size=VALIDATION_BATCH_SIZE,
net_cls=net_cls,
data_n=df.at[0, 'total_valid_data'],
)
# ========== ネットワーク呼び出し ==========
fit_cls = FitManager(
net_cls=net_cls,
path_cls=path_cls,
shuf_train_ds_cls=shuf_train_ds_cls,
train_ds_cls=train_ds_cls,
test_ds_cls=test_ds_cls,
validation_ds_cls=valid_ds_cls,
check_img_path=check_img_path,
