Merge pull request #159 from SamitHuang/version

fix setup import

Author: SamitHuang

__init__.py

@@ -1 +1,8 @@
+from . import mindocr
+
 from .mindocr import *
+from .mindocr import data, losses, metrics, models, postprocess, utils
+
+__all__ = []
+__all__.extend(mindocr.__all__)
+__all__.extend(['data', 'losses', 'metrics', 'models', 'postprocess', 'utils'])

mindocr/data/transforms/general_transforms.py

@@ -4,7 +4,7 @@
 from PIL import Image
 from mindspore.dataset.vision import RandomColorAdjust as MSRandomColorAdjust
 
-from mindocr.data.constants import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
+from ...data.constants import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
 
 __all__ = ['DecodeImage', 'NormalizeImage', 'ToCHWImage', 'PackLoaderInputs', 'ScalePadImage', 'GridResize',
            'RandomScale', 'RandomCropWithBBox', 'RandomColorAdjust']

mindocr/utils/callbacks.py

@@ -8,8 +8,8 @@
 import mindspore as ms
 from mindspore import save_checkpoint
 from mindspore.train.callback._callback import Callback, _handle_loss
-from mindocr.utils.visualize import draw_bboxes, show_imgs, recover_image
-from mindocr.utils.recorder import PerfRecorder
+from .visualize import draw_bboxes, show_imgs, recover_image
+from .recorder import PerfRecorder
 
 __all__ = ['Evaluator', 'EvalSaveCallback']
 

mindocr/utils/visualize.py

@@ -2,7 +2,7 @@
 import numpy as np
 from typing import Union, List
 import matplotlib.pyplot as plt
-from mindocr.data.constants import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
+from ..data.constants import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
 
 __all__ = ['show_img', 'show_imgs', 'draw_bboxes', 'recover_image']