from .color2normal_dataset import Color2NormalDataset from .mlp_model import MLP