data_generator() is not defined in magnet.py

[Bob-cheng]

On the branch "defense_autoencoder"
In the file car_motion_attack/magnet.py line 116, the function data_generator() is not defined.
I cannot find the function anywhere in this repository. Could you please have a look and update the missing code?

Thanks a lot.

[tkm2261]

Thank you for your report! I accidentally dropped a function when clearing up the code.

The following code is the data_generator(). I will also update the branch as well. Please let me know if it does not work.

Best,

Takami

def data_generator(ids_train_split, batch_size):
    while True:
        _ids_train_split = np.random.permutation(ids_train_split)
        for start in range(0, len(_ids_train_split), batch_size):  # , ncols=100, desc="loader"):
            x_batch = []
            end = min(start + batch_size, len(_ids_train_split))
            ids_train_batch = _ids_train_split[start:end]

            for id in ids_train_batch:
                img = cv2.cvtColor(cv2.imread(id), cv2.COLOR_BGR2RGB)
                img = cv2.warpPerspective(img, ROI_MAT_INV,
                                          (IMG_CROP_WIDTH, IMG_CROP_HEIGHT))
                x_batch.append(img)

            x_batch = np.stack(x_batch).astype(np.float16) / 255

            yield x_batch, x_batch