소스 검색

Update farseg.py

Fixed the bug of ‘import paddlers’
kongdebug 3 년 전
부모
커밋
faf4d1962f
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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]