demo.py updated

Yemini-Lab · Jun 22, 2020 · bd90896 · bd90896
1 parent aa67376
commit bd90896
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -136,3 +136,9 @@ dmypy.json
 
 # Cython debug symbols
 cython_debug/
+
+
+# images and videos
+*.mp4
+*.png
+*.jpg
diff --git a/demo.py b/demo.py
@@ -47,16 +47,20 @@
 A = torch.tensor(A).float()
 n_nmf = dNMF(torch.tensor(m.mc[0]).float(),positions=A[:,:,np.newaxis],\
             radius=params['radius'],step_S=params['step_S'],gamma=params['gamma'],
-            use_gpu=False,initial_p=A,sigma_inv=params['sigma_inv'], method='1->t', verbose=False)
+            use_gpu=False,initial_p=A,sigma_inv=params['sigma_inv'],
+            method='1->t', verbose=True)
 n_nmf.optimize(lr=.1,n_iter=0,n_iter_c=20)
 end = time.time()
 print('normcorre-nmf finished in ' + str(end-start) + ' seconds')
 
 start = time.time()
 dnmf = dNMF(video,positions=positions[:,:,0][:,:,np.newaxis],\
     radius=params['radius'],step_S=params['step_S'],gamma=params['gamma'],
-    use_gpu=False,initial_p=positions[:,:,0],sigma_inv=params['sigma_inv'], method='1->t', verbose=False)
+    use_gpu=False,initial_p=positions[:,:,0],sigma_inv=params['sigma_inv'],
+    method='1->t', verbose=True)
 
-dnmf.optimize(lr=.1,n_iter=20,n_iter_c=2)
+dnmf.optimize(lr=1e-4,n_iter=20,n_iter_c=2)
 end = time.time()
-print('dNMF finished in ' + str(end-start) + ' seconds')
+print('dNMF finished in ' + str(end-start) + ' seconds')
+
+dnmf.visualize_tracks('result',video)