Browse Source

Update farseg.py

Fixed the bug of ‘import paddlers’
kongdebug 3 years ago
parent
commit
faf4d1962f
1 changed files with 2 additions and 2 deletions
  1. 2 2
      paddlers/custom_models/seg/models/farseg/farseg.py

+ 2 - 2
paddlers/custom_models/seg/models/farseg/farseg.py

@@ -17,7 +17,7 @@ import paddle.nn as nn
 import paddle.nn.functional as F
 from paddle.vision.models import resnet50
 from .fpn import FPN
-from seg.models.utils import Identity
+from ..utils import Identity
 
 
 class SceneRelation(nn.Layer):
@@ -168,4 +168,4 @@ class FarSeg(nn.Layer):
         cls_pred = self.cls_pred_conv(final_feat)
         cls_pred = self.upsample4x_op(cls_pred)
         cls_pred = F.softmax(cls_pred, axis=1)
-        return [cls_pred]
+        return [cls_pred]