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