update gaitedge and use data_in_use

2022-07-19 17:34:20 +08:00
parent 4b681fb9bd
commit 8edfc80326
7 changed files with 66 additions and 24 deletions
@@ -15,9 +15,8 @@ class Segmentation(BaseModel):
    def forward(self, inputs):
        ipts, labs, typs, vies, seqL = inputs
        del seqL
-        # ratios = ipts[0]
-        rgbs = ipts[1]
-        sils = ipts[2]
+        rgbs = ipts[0]
+        sils = ipts[1]
        # del ipts
        n, s, c, h, w = rgbs.size()
        rgbs = rgbs.view(n*s, c, h, w)