Commit 3b33e917 authored by Gabriel Kirsten's avatar Gabriel Kirsten

added pseudo label

parent d424033a
...@@ -23,6 +23,12 @@ except Exception as e: ...@@ -23,6 +23,12 @@ except Exception as e:
CNNKeras = None CNNKeras = None
print e.message print e.message
try:
from .cnn_pseudo_label_keras import CNNPseudoLabel
except Exception as e:
CNNPseudoLabel = None
print e.message
try: try:
from .segnet_keras import SEGNETKeras from .segnet_keras import SEGNETKeras
...@@ -34,6 +40,7 @@ except Exception as e: ...@@ -34,6 +40,7 @@ except Exception as e:
__all__ = ["classifier", __all__ = ["classifier",
"cnn_caffe", "cnn_caffe",
"cnn_keras", "cnn_keras",
"cnn_pseudo_label_keras",
"segnet_keras", "segnet_keras",
"weka_classifiers", "weka_classifiers",
"syntactic" "syntactic"
...@@ -49,6 +56,8 @@ _classifier_list = OrderedDict( [ ...@@ -49,6 +56,8 @@ _classifier_list = OrderedDict( [
WekaClassifiers is None and CNNCaffe is not None, bool, meta=CNNCaffe, hidden=CNNCaffe is None)], WekaClassifiers is None and CNNCaffe is not None, bool, meta=CNNCaffe, hidden=CNNCaffe is None)],
["cnn_keras", Config("Invalid" if CNNKeras is None else CNNKeras.__name__, ["cnn_keras", Config("Invalid" if CNNKeras is None else CNNKeras.__name__,
CNNKeras is not None, bool, meta=CNNKeras, hidden=CNNKeras is None)], CNNKeras is not None, bool, meta=CNNKeras, hidden=CNNKeras is None)],
["cnn_pseudo_label_keras", Config("Invalid" if CNNPseudoLabel is None else CNNPseudoLabel.__name__,
CNNPseudoLabel is not None, bool, meta=CNNPseudoLabel, hidden=CNNPseudoLabel is None)],
["segnet_keras", Config("Invalid" if SEGNETKeras is None else SEGNETKeras.__name__, ["segnet_keras", Config("Invalid" if SEGNETKeras is None else SEGNETKeras.__name__,
SEGNETKeras is not None, bool, meta=SEGNETKeras, hidden=SEGNETKeras is None)], SEGNETKeras is not None, bool, meta=SEGNETKeras, hidden=SEGNETKeras is None)],
["weka_classifiers", Config("Invalid" if WekaClassifiers is None else WekaClassifiers.__name__, ["weka_classifiers", Config("Invalid" if WekaClassifiers is None else WekaClassifiers.__name__,
...@@ -63,6 +72,7 @@ def get_classifier_config(): ...@@ -63,6 +72,7 @@ def get_classifier_config():
def set_classifier_config(configs): def set_classifier_config(configs):
_classifier_list["cnn_caffe"] = Config.nvl_config(configs["cnn_caffe"], _classifier_list["cnn_caffe"]) _classifier_list["cnn_caffe"] = Config.nvl_config(configs["cnn_caffe"], _classifier_list["cnn_caffe"])
_classifier_list["cnn_keras"] = Config.nvl_config(configs["cnn_keras"], _classifier_list["cnn_keras"]) _classifier_list["cnn_keras"] = Config.nvl_config(configs["cnn_keras"], _classifier_list["cnn_keras"])
_classifier_list["cnn_pseudo_label_keras"] = Config.nvl_config(configs["cnn_pseudo_label_keras"], _classifier_list["cnn_pseudo_label_keras"])
_classifier_list["segnet_keras"] = Config.nvl_config(configs["segnet_keras"], _classifier_list["segnet_keras"]) _classifier_list["segnet_keras"] = Config.nvl_config(configs["segnet_keras"], _classifier_list["segnet_keras"])
_classifier_list["weka_classifiers"] = Config.nvl_config(configs["weka_classifiers"], _classifier_list["weka_classifiers"]) _classifier_list["weka_classifiers"] = Config.nvl_config(configs["weka_classifiers"], _classifier_list["weka_classifiers"])
_classifier_list["syntactic"] = Config.nvl_config(configs["syntactic"], _classifier_list["syntactic"]) _classifier_list["syntactic"] = Config.nvl_config(configs["syntactic"], _classifier_list["syntactic"])
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment