Browse Source

试一试

master
liupengfei 1 month ago
parent
commit
67c415308e
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/models.py

+ 2
- 2
src/models.py View File

@@ -119,8 +119,8 @@ class FlowNet2(nn.Cell):

# flownetsd
flownetsd_flow2 = self.flownets_d(x)[0]
# up3in = flownetsd_flow2 / self.div_flow
up3in = flownetsd_flow2
up3in = flownetsd_flow2 / self.div_flow
# up3in = flownetsd_flow2
flownetsd_flow = self.upsample3(up3in)
norm_flownetsd_flow = self.channelnorm(flownetsd_flow)



Loading…
Cancel
Save