continue fixing linter

Author: kprokofi

torchdet3d/builders/loader_builder.py

@@ -3,7 +3,7 @@
 import cv2
 
 from torchdet3d.dataloaders import Objectron
-import torchdet3d.utils as uti
+from torchdet3d.utils import ConvertColor, ToTensor
 
 def build_loader(config, mode='train'):
 
@@ -30,20 +30,20 @@ def build_augmentations(cfg):
     normalize = A.augmentations.transforms.Normalize (**cfg.data.normalization)
 
     train_transform = A.Compose([
-                            uti.ConvertColor(),
+                            ConvertColor(),
                             A.Resize(*cfg.data.resize, interpolation=cv2.INTER_CUBIC),
                             A.HorizontalFlip(p=0.3),
                             # A.Rotate(limit=30, p=0.3, interpolation=cv2.INTER_CUBIC),
                             A.RandomBrightnessContrast(p=0.2),
                             normalize,
-                            uti.ToTensor(cfg.data.resize)
+                            ToTensor(cfg.data.resize)
                             ], keypoint_params=A.KeypointParams(format='xy', remove_invisible=False))
 
     test_transform = A.Compose([
-                            uti.ConvertColor(),
+                            ConvertColor(),
                             A.Resize(*cfg.data.resize,cv2.INTER_CUBIC),
                             normalize,
-                            uti.ToTensor(cfg.data.resize)
+                            ToTensor(cfg.data.resize)
                             ], keypoint_params=A.KeypointParams(format='xy', remove_invisible=False))
 
     return train_transform, test_transform

torchdet3d/builders/model_builder.py

@@ -11,7 +11,7 @@
 from efficientnet_lite2_pytorch_model import EfficientnetLite2ModelFile
 
 from torchdet3d.models import MobileNetV3, init_pretrained_weights, model_params
-import torchdet3d.utils as uti
+from torchdet3d.utils import load_pretrained_weights
 
 __AVAI_MODELS__ = {
                     'mobilenetv3_large', 'mobilenetv3_small', 'efficientnet-lite0', 'efficientnet-lite1',
@@ -37,7 +37,7 @@ def build_model(config, export_mode=False):
                              global_params=global_params)
 
         if config.model.pretrained and not export_mode:
-            uti.load_pretrained_weights(model, weights_path)
+            load_pretrained_weights(model, weights_path)
 
     elif config.model.name == 'mobilenetv3_large':
         params = model_params['mobilenetv3_large']

torchdet3d/utils/transforms.py

@@ -3,7 +3,7 @@
 import torch
 from albumentations.core.transforms_interface import BasicTransform, ImageOnlyTransform
 
-from torchdet3d.utils import normalize
+from utils import normalize
 
 class ConvertColor(ImageOnlyTransform):
     """Converting color of the image