浏览代码

Merge pull request #68 from Ichaelus/ichaelus/fix-import-statement

Fix a import statement that prevents deoldify functions to be called from other folders
Jason Antic 6 年之前
父节点
当前提交
334030a098
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      fasterai/visualize.py

+ 1 - 1
fasterai/visualize.py

@@ -7,7 +7,7 @@ from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
 from fastai.dataset import FilesDataset, ImageData, ModelData, open_image
 from fastai.dataset import FilesDataset, ImageData, ModelData, open_image
 from fastai.transforms import Transform, scale_min, tfms_from_stats, inception_stats
 from fastai.transforms import Transform, scale_min, tfms_from_stats, inception_stats
 from fastai.transforms import CropType, NoCrop, Denormalize, Scale
 from fastai.transforms import CropType, NoCrop, Denormalize, Scale
-from fasterai.transforms import BlackAndWhiteTransform
+from .transforms import BlackAndWhiteTransform
 from .training import GenResult, CriticResult, GANTrainer
 from .training import GenResult, CriticResult, GANTrainer
 from .images import ModelImageSet, EasyTensorImage
 from .images import ModelImageSet, EasyTensorImage
 from .generators import GeneratorModule
 from .generators import GeneratorModule