Predicting batch of image arrays

[deeperlearner]

Pytorch-UNet/predict.py

Lines 15 to 19 in 21d7850

	def predict_img(net,
	full_img,
	device,
	scale_factor=1,
	out_threshold=0.5):

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).