python - 有没有办法在 MobileNetV2 上实现我自己的功能作为预处理？

Question

我正在考虑这样的事情，在其中定义我想要的函数，当我生成数据集时，过滤器将应用于该数据集中的每个图像：

def MyFunction(Image):
  # do something to every image
  return new_image

train_dataset = image_dataset_from_directory(train_dir,
                                             shuffle=True,
                                             batch_size=BATCH_SIZE,
                                             image_size=IMG_SIZE,
                                             **pre_process = MyFunction**)

标签： pythonmachine-learningimage-processingimage-classificationmobilenet