Mercurial > repos > bgruening > bioimage_inference

comparison main.py @ 4:2b61d8fcfa52 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/bioimaging commit f038722c21eaa3018e1cef0004ac7bd283d15269
author bgruening
date Mon, 07 Apr 2025 14:46:13 +0000
parents bc28236f407b
children 37b9ead209da
comparison
equal deleted inserted replaced
3:bc28236f407b 4:2b61d8fcfa52
139 # post process predicted file to correctly save as TIF file 139 # post process predicted file to correctly save as TIF file
140 pred_data = torch.squeeze(pred_data) 140 pred_data = torch.squeeze(pred_data)
141 pred_numpy = pred_data.detach().numpy() 141 pred_numpy = pred_data.detach().numpy()
142 142
143 # write predicted TIF image to file 143 # write predicted TIF image to file
144 imageio.v3.imwrite("output_predicted_image.tif", pred_numpy, extension=".tif") 144 imageio.v3.imwrite("output_predicted_image.tiff", pred_numpy, extension=".tiff")