update

HVision-NKU · Dec 9, 2022 · 7f5b855 · 7f5b855
1 parent 3150da8
commit 7f5b855
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -114,7 +114,7 @@ def train(Dataset, parser):
                 path=_TESTDATASET_+'/'+path
                 t = Valid(dataset, path, epoch, 'CamoFormer','output/checkpoint/CamoFormer/CamoFormer/' )
                 t.save()
-            os.system('bash evaltools/valid_eval.sh '+'output/Prediction/CamoFormer-epoch'+str(epoch+1)+' '+_TESTDATASET_)
+            os.system('bash evaltools/valid_eval.sh '+'output/Prediction/CamoFormer-epoch'+str(epoch+1)+' '+_TESTDATASET_+' CamoFormer'+str(epoch+1))
 
 def test(dataset,parser):
     args   = parser.parse_args()

diff --git a/test_eval.sh b/test_eval.sh
@@ -1,5 +1,5 @@
 python main.py \
-    --ckpt 'checkpoint/CamoFormer56'\
+    --ckpt 'checkpoint/CamoFormer-trained'\
     --mode 'test'
 
 python evaltools/eval.py   \