Pytorch-UNet icon indicating copy to clipboard operation
Pytorch-UNet copied to clipboard

Published 20 hours ago • verified publisher icon milesial

Predicting batch of image arrays

Open deeperlearner opened this issue 1 year ago • 0 comments

https://github.com/milesial/Pytorch-UNet/blob/21d7850f2af30a9695bbeea75f3136aa538cfc4a/predict.py#L15-L19 This function seems to predict a PIL image a time.

How can predict_img process a batch of images. For example, full_imgs = np.random.rand(batch_size, 3, 112, 112).

deeperlearner avatar May 29 '24 07:05 deeperlearner