Update test.py

ohhhyeahhh · Jul 2, 2020 · c8b2120 · c8b2120
1 parent cd0bd73
commit c8b2120
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tools/test.py b/tools/test.py
@@ -13,6 +13,7 @@
 import numpy as np
 import math
 import sys
+import shutil
 sys.path.append('../')
 
 from pysot.core.config import cfg
@@ -115,6 +116,10 @@ def main():
                 f.write(','.join([str(i) for i in x])+'\n')
         print('({:3d}) Video: {:12s} Time: {:5.1f}s Speed: {:3.1f}fps'.format(
             v_idx+1, video.name, toc, idx / toc))
+    os.chdir(model_path)
+    save_file = '../%s' % dataset
+    shutil.make_archive(save_file, 'zip')
+    print('Records saved at', save_file + '.zip')
 
 
 if __name__ == '__main__':