|
|
@@ -119,7 +119,8 @@ class FlowNet2(nn.Cell): |
|
|
|
|
|
|
|
# flownetsd |
|
|
|
flownetsd_flow2 = self.flownets_d(x)[0] |
|
|
|
up3in = flownetsd_flow2 / self.div_flow |
|
|
|
# up3in = flownetsd_flow2 / self.div_flow |
|
|
|
up3in = flownetsd_flow2 |
|
|
|
flownetsd_flow = self.upsample3(up3in) |
|
|
|
norm_flownetsd_flow = self.channelnorm(flownetsd_flow) |
|
|
|
|
|
|
|