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.transforms import Transform, scale_min, tfms_from_stats, inception_stats
 from fastai.transforms import CropType, NoCrop, Denormalize, Scale
-from fasterai.transforms import BlackAndWhiteTransform
+from .transforms import BlackAndWhiteTransform
 from .training import GenResult, CriticResult, GANTrainer
 from .images import ModelImageSet, EasyTensorImage
 from .generators import GeneratorModule