Fix for bug #6

shalinshah1993 · Feb 3, 2020 · 58f09ea · 58f09ea
1 parent 92bd50b
commit 58f09ea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/findalllocalizations.m b/findalllocalizations.m
@@ -44,7 +44,7 @@ function findalllocalizations(file, threshold, ...
             % number of localizations in each frame
             localizations(iFrame) = size(points, 1);
 
-            if toDisplayFrames
+            if toDisplayFrames && ~sum(isnan(points(:)))
                 figure(1);          
                 localized_frame = insertShape(cur_frame,'circle', [points(:, 1) ...
                     points(:, 2) 3.*ones(length(points(:, 1)), 1)], 'LineWidth', 1);
@@ -75,4 +75,4 @@ function findalllocalizations(file, threshold, ...
     end
     save(strcat('tmp/all_pnts/', fileName{1}), 'localizationList', ...
                                             'vidSize','-v7.3');
-end
+end