XSeg: added pretrain option.

2021-07-30 17:24:21 +04:00
parent 83b1412da7
commit 55b947eab5
3 changed files with 111 additions and 46 deletions
@@ -81,8 +81,8 @@ class XSegNet(object):
    def get_resolution(self):
        return self.resolution
        
-    def flow(self, x):
-        return self.model(x)
+    def flow(self, x, pretrain=False):
+        return self.model(x, pretrain=pretrain)

    def get_weights(self):
        return self.model_weights