update process_template.py

ChristophKirst · Jul 19, 2016 · 16fd455 · 16fd455
1 parent 7b14c2c
commit 16fd455
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ClearMap/Scripts/process_template.py b/ClearMap/Scripts/process_template.py
@@ -74,12 +74,14 @@
 
 #Without weigths:
 vox = voxelize(points, AtlasFile, **voxelizeParameter);
-io.writeData(os.path.join(BaseDirectory, 'cells_heatmap.tif'), vox.astype('int32'));
+if not isinstance(vox, basestring):
+  io.writeData(os.path.join(BaseDirectory, 'cells_heatmap.tif'), vox.astype('int32'));
 
 #With weigths from the intensity file (here raw intensity):
 voxelizeParameter["weights"] = intensities[:,0].astype(float);
 vox = voxelize(points, AtlasFile, **voxelizeParameter);
-io.writeData(os.path.join(BaseDirectory, 'cells_heatmap_weighted.tif'), vox.astype('int32'));
+if not isinstance(vox, basestring):
+  io.writeData(os.path.join(BaseDirectory, 'cells_heatmap_weighted.tif'), vox.astype('int32'));