update

cvlab-yonsei · Aug 13, 2020 · ea96458 · ea96458
1 parent c3413d8
commit ea96458
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/Evaluate.py b/Evaluate.py
@@ -51,6 +51,8 @@
 
 args = parser.parse_args()
 
+torch.manual_seed(2020)
+
 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
 if args.gpus is None:
     gpus = "0"
@@ -81,7 +83,11 @@
 model = torch.load(args.model_dir)
 model.cuda()
 m_items = torch.load(args.m_items_dir)
+
+
 labels = np.load('./data/frame_labels_'+args.dataset_type+'.npy')
+if args.dataset_type == 'shanghai':
+    labels = np.expand_dims(labels, 0)
 
 videos = OrderedDict()
 videos_list = sorted(glob.glob(os.path.join(test_folder, '*')))

diff --git a/README.md b/README.md
@@ -7,7 +7,7 @@ For more information, checkout the project site [[website](https://cvlab.yonsei.
 
 ## Dependencies
 * Python 3.6
-* PyTorch >= 1.0.0
+* PyTorch 1.1.0
 * Numpy
 * Sklearn
 

diff --git a/Train.py b/Train.py
@@ -52,6 +52,8 @@
 
 args = parser.parse_args()
 
+torch.manual_seed(2020)
+
 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
 if args.gpus is None:
     gpus = "0"