From ec2093594cd52dca462aa6a57c298221b7c0116f Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Fri, 5 Oct 2018 17:17:48 +0800
Subject: [PATCH] bug fix
---
mmdet/core/losses/__init__.py | 13 ++++++-------
1 file changed, 6 insertions(+), 7 deletions(-)
diff --git a/mmdet/core/losses/__init__.py b/mmdet/core/losses/__init__.py
index 3e4447f..661f0d6 100644
--- a/mmdet/core/losses/__init__.py
+++ b/mmdet/core/losses/__init__.py
@@ -1,12 +1,11 @@
-from .losses import (
- weighted_nll_loss, weighted_cross_entropy, weighted_binary_cross_entropy,
- sigmoid_focal_loss, weighted_sigmoid_focal_loss, mask_cross_entropy,
- weighted_mask_cross_entropy, smooth_l1_loss, weighted_smoothl1, accuracy)
+from .losses import (weighted_nll_loss, weighted_cross_entropy,
+ weighted_binary_cross_entropy, sigmoid_focal_loss,
+ weighted_sigmoid_focal_loss, mask_cross_entropy,
+ smooth_l1_loss, weighted_smoothl1, accuracy)
__all__ = [
'weighted_nll_loss', 'weighted_cross_entropy',
'weighted_binary_cross_entropy', 'sigmoid_focal_loss',
- 'weighted_sigmoid_focal_loss', 'mask_cross_entropy',
- 'weighted_mask_cross_entropy', 'smooth_l1_loss', 'weighted_smoothl1',
- 'accuracy'
+ 'weighted_sigmoid_focal_loss', 'mask_cross_entropy', 'smooth_l1_loss',
+ 'weighted_smoothl1', 'accuracy'
]
--
GitLab