xpngzhng

from keras.layers import Activation, Reshape, Lambda, dot, add
from keras.layers import Conv1D, Conv2D, Conv3D
from keras.layers import MaxPool1D, MaxPool2D, MaxPool3D
from keras import backend as K
import tensorflow as tf


def non_local_block(ip, intermediate_dim=None, compression=2,
                    mode='embedded', add_residual=True, prefix=None):
    """

xpngzhng

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

from keras.preprocessing.image import ImageDataGenerator
from keras import callbacks

train_dir = 'D:\\Meitu\\Slim\\train'
validation_dir = 'D:\\Meitu\\Slim\\val'
save_dir = './result'
image_size = 224