Browse Source

Fix a import statement that prevents deoldify functions to be called from other folders

Ichaelus 6 years ago
parent
commit
ac1ef9ffcf
1 changed files with 1 additions and 1 deletions
  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