Unverified Commit d2483e15 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Use isort to sort imports and setup travis (#1085)

* add isort config

* use isort to sort imports

* add isort to travis
parent 864880de
from .two_stage import TwoStageDetector
from ..registry import DETECTORS from ..registry import DETECTORS
from .two_stage import TwoStageDetector
@DETECTORS.register_module @DETECTORS.register_module
......
import torch import torch
from mmdet.core import bbox2roi, build_assigner, build_sampler from mmdet.core import bbox2roi, build_assigner, build_sampler
from .two_stage import TwoStageDetector
from .. import builder from .. import builder
from ..registry import DETECTORS from ..registry import DETECTORS
from .two_stage import TwoStageDetector
@DETECTORS.register_module @DETECTORS.register_module
......
from .single_stage import SingleStageDetector
from ..registry import DETECTORS from ..registry import DETECTORS
from .single_stage import SingleStageDetector
@DETECTORS.register_module @DETECTORS.register_module
......
import mmcv import mmcv
from mmdet.core import tensor2imgs, bbox_mapping from mmdet.core import bbox_mapping, tensor2imgs
from .base import BaseDetector
from .test_mixins import RPNTestMixin
from .. import builder from .. import builder
from ..registry import DETECTORS from ..registry import DETECTORS
from .base import BaseDetector
from .test_mixins import RPNTestMixin
@DETECTORS.register_module @DETECTORS.register_module
......
import torch.nn as nn import torch.nn as nn
from .base import BaseDetector from mmdet.core import bbox2result
from .. import builder from .. import builder
from ..registry import DETECTORS from ..registry import DETECTORS
from mmdet.core import bbox2result from .base import BaseDetector
@DETECTORS.register_module @DETECTORS.register_module
......
from mmdet.core import (bbox2roi, bbox_mapping, merge_aug_proposals, from mmdet.core import (bbox2roi, bbox_mapping, merge_aug_bboxes,
merge_aug_bboxes, merge_aug_masks, multiclass_nms) merge_aug_masks, merge_aug_proposals, multiclass_nms)
class RPNTestMixin(object): class RPNTestMixin(object):
......
import torch import torch
import torch.nn as nn import torch.nn as nn
from .base import BaseDetector from mmdet.core import bbox2result, bbox2roi, build_assigner, build_sampler
from .test_mixins import RPNTestMixin, BBoxTestMixin, MaskTestMixin
from .. import builder from .. import builder
from ..registry import DETECTORS from ..registry import DETECTORS
from mmdet.core import bbox2roi, bbox2result, build_assigner, build_sampler from .base import BaseDetector
from .test_mixins import BBoxTestMixin, MaskTestMixin, RPNTestMixin
@DETECTORS.register_module @DETECTORS.register_module
......
...@@ -2,8 +2,8 @@ import numpy as np ...@@ -2,8 +2,8 @@ import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
from .utils import weighted_loss
from ..registry import LOSSES from ..registry import LOSSES
from .utils import weighted_loss
@weighted_loss @weighted_loss
......
...@@ -2,8 +2,8 @@ import torch ...@@ -2,8 +2,8 @@ import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from .utils import weight_reduce_loss
from ..registry import LOSSES from ..registry import LOSSES
from .utils import weight_reduce_loss
def cross_entropy(pred, label, weight=None, reduction='mean', avg_factor=None): def cross_entropy(pred, label, weight=None, reduction='mean', avg_factor=None):
......
...@@ -2,8 +2,8 @@ import torch.nn as nn ...@@ -2,8 +2,8 @@ import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from mmdet.ops import sigmoid_focal_loss as _sigmoid_focal_loss from mmdet.ops import sigmoid_focal_loss as _sigmoid_focal_loss
from .utils import weight_reduce_loss
from ..registry import LOSSES from ..registry import LOSSES
from .utils import weight_reduce_loss
# This method is only for debugging # This method is only for debugging
......
...@@ -2,8 +2,8 @@ import torch ...@@ -2,8 +2,8 @@ import torch
import torch.nn as nn import torch.nn as nn
from mmdet.core import bbox_overlaps from mmdet.core import bbox_overlaps
from .utils import weighted_loss
from ..registry import LOSSES from ..registry import LOSSES
from .utils import weighted_loss
@weighted_loss @weighted_loss
......
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from .utils import weighted_loss
from ..registry import LOSSES from ..registry import LOSSES
from .utils import weighted_loss
mse_loss = weighted_loss(F.mse_loss) mse_loss = weighted_loss(F.mse_loss)
......
import torch import torch
import torch.nn as nn import torch.nn as nn
from .utils import weighted_loss
from ..registry import LOSSES from ..registry import LOSSES
from .utils import weighted_loss
@weighted_loss @weighted_loss
......
...@@ -4,10 +4,10 @@ import pycocotools.mask as mask_util ...@@ -4,10 +4,10 @@ import pycocotools.mask as mask_util
import torch import torch
import torch.nn as nn import torch.nn as nn
from mmdet.core import auto_fp16, force_fp32, mask_target
from ..builder import build_loss from ..builder import build_loss
from ..registry import HEADS from ..registry import HEADS
from ..utils import ConvModule from ..utils import ConvModule
from mmdet.core import mask_target, force_fp32, auto_fp16
@HEADS.register_module @HEADS.register_module
......
from .fcn_mask_head import FCNMaskHead
from ..registry import HEADS from ..registry import HEADS
from ..utils import ConvModule from ..utils import ConvModule
from .fcn_mask_head import FCNMaskHead
@HEADS.register_module @HEADS.register_module
......
...@@ -2,8 +2,8 @@ import numpy as np ...@@ -2,8 +2,8 @@ import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
from mmcv.cnn import kaiming_init, normal_init from mmcv.cnn import kaiming_init, normal_init
from mmdet.core import force_fp32
from mmdet.core import force_fp32
from ..builder import build_loss from ..builder import build_loss
from ..registry import HEADS from ..registry import HEADS
......
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from torch.utils.checkpoint import checkpoint
from mmcv.cnn.weight_init import caffe2_xavier_init from mmcv.cnn.weight_init import caffe2_xavier_init
from torch.utils.checkpoint import checkpoint
from ..utils import ConvModule
from ..registry import NECKS from ..registry import NECKS
from ..utils import ConvModule
@NECKS.register_module @NECKS.register_module
......
import math
import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
import math
import numpy as np
from mmcv.cnn import kaiming_init from mmcv.cnn import kaiming_init
......
import warnings import warnings
import torch.nn as nn import torch.nn as nn
from mmcv.cnn import kaiming_init, constant_init from mmcv.cnn import constant_init, kaiming_init
from .conv_ws import ConvWS2d from .conv_ws import ConvWS2d
from .norm import build_norm_layer from .norm import build_norm_layer
......
import numpy as np import numpy as np
import torch import torch
from . import nms_cuda, nms_cpu from . import nms_cpu, nms_cuda
from .soft_nms_cpu import soft_nms_cpu from .soft_nms_cpu import soft_nms_cpu
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment