Add files via upload

tntek · Apr 13, 2024 · e7b7223 · e7b7223
1 parent fa5cc32
commit e7b7223
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/image_target_of_oh_vs.py b/image_target_of_oh_vs.py
@@ -12,6 +12,8 @@
 import src.methods.oh.nrc_vs as Nrc_vs
 import src.methods.oh.lcfd as LCFD
 import src.methods.oh.difo as DIFO
+import src.methods.oh.difo_50 as DIFO_50
+
 import src.methods.oh.plue as PLUE
 import src.methods.oh.adacontrast as ADACONTRAST
 import src.methods.oh.source as SOURCE
@@ -77,7 +79,10 @@
 
     elif cfg.MODEL.METHOD == "difo":
         print("using difo method")
-        acc = DIFO.train_target(cfg)
+        if 'RN' in cfg.DIFO.ARCH:
+            acc = DIFO_50.train_target(cfg)
+        else:
+            acc = DIFO.train_target(cfg)
 
     elif cfg.MODEL.METHOD == "plue":
         print("using plue method")