Browse Source

122行不得不改

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

+ 2
- 1
src/models.py View File

@@ -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)



Loading…
Cancel
Save