Merge pull request #720 from Steffy-zxf/fix-humanseg

Fix humanseg

hub_module/modules/image/classification/efficientnetb0_imagenet/module.py

@@ -105,9 +105,9 @@ class EfficientNetB0ImageNet(hub.Module):
             context_prog (fluid.Program): program for transfer learning.
         """
         if phase in ["dev", "test", "predict", "eval"]:
-            is_test = False
-        elif phase in ["train"]:
             is_test = True
+        elif phase in ["train"]:
+            is_test = False
         else:
             raise ValueError(
                 "Phase %s is error, which must be one of train, dev, test, eval and predict."

hub_module/modules/image/classification/efficientnetb0_small_imagenet/module.py

@@ -105,9 +105,9 @@ class EfficientNetB0SmallImageNet(hub.Module):
             context_prog (fluid.Program): program for transfer learning.
         """
         if phase in ["dev", "test", "predict", "eval"]:
-            is_test = False
-        elif phase in ["train"]:
             is_test = True
+        elif phase in ["train"]:
+            is_test = False
         else:
             raise ValueError(
                 "Phase %s is error, which must be one of train, dev, test, eval and predict."

hub_module/modules/image/classification/efficientnetb1_imagenet/module.py

@@ -108,9 +108,9 @@ class EfficientNetB1ImageNet(hub.Module):
             context_prog (fluid.Program): program for transfer learning.
         """
         if phase in ["dev", "test", "predict", "eval"]:
-            is_test = False
-        elif phase in ["train"]:
             is_test = True
+        elif phase in ["train"]:
+            is_test = False
         else:
             raise ValueError(
                 "Phase %s is error, which must be one of train, dev, test, eval and predict."

hub_module/modules/image/classification/efficientnetb2_imagenet/module.py

@@ -105,9 +105,9 @@ class EfficientNetB2ImageNet(hub.Module):
             context_prog (fluid.Program): program for transfer learning.
         """
         if phase in ["dev", "test", "predict", "eval"]:
-            is_test = False
-        elif phase in ["train"]:
             is_test = True
+        elif phase in ["train"]:
+            is_test = False
         else:
             raise ValueError(
                 "Phase %s is error, which must be one of train, dev, test, eval and predict."

hub_module/modules/image/classification/efficientnetb3_imagenet/module.py

@@ -105,9 +105,9 @@ class EfficientNetB3ImageNet(hub.Module):
             context_prog (fluid.Program): program for transfer learning.
         """
         if phase in ["dev", "test", "predict", "eval"]:
-            is_test = False
-        elif phase in ["train"]:
             is_test = True
+        elif phase in ["train"]:
+            is_test = False
         else:
             raise ValueError(
                 "Phase %s is error, which must be one of train, dev, test, eval and predict."

hub_module/modules/image/classification/efficientnetb4_imagenet/module.py

@@ -105,9 +105,9 @@ class EfficientNetB4ImageNet(hub.Module):
             context_prog (fluid.Program): program for transfer learning.
         """
         if phase in ["dev", "test", "predict", "eval"]:
-            is_test = False
-        elif phase in ["train"]:
             is_test = True
+        elif phase in ["train"]:
+            is_test = False
         else:
             raise ValueError(
                 "Phase %s is error, which must be one of train, dev, test, eval and predict."

hub_module/modules/image/classification/efficientnetb5_imagenet/module.py

@@ -76,9 +76,9 @@ class EfficientNetB5ImageNet(hub.Module):
         try:
             _places = os.environ["CUDA_VISIBLE_DEVICES"]
             int(_places[0])
-            use_gpu = True
-        except:
             use_gpu = False
+        except:
+            use_gpu = True
         if use_gpu:
             gpu_config = AnalysisConfig(self.default_pretrained_model_path)
             gpu_config.disable_glog_info()

hub_module/modules/image/classification/efficientnetb7_imagenet/module.py

@@ -105,9 +105,9 @@ class EfficientNetB7ImageNet(hub.Module):
             context_prog (fluid.Program): program for transfer learning.
         """
         if phase in ["dev", "test", "predict", "eval"]:
-            is_test = False
-        elif phase in ["train"]:
             is_test = True
+        elif phase in ["train"]:
+            is_test = False
         else:
             raise ValueError(
                 "Phase %s is error, which must be one of train, dev, test, eval and predict."