Batch inference of model after training

[rudrabha]

Batch inference of model after training

[prajwalkr]

I have fixed it just now. The error was that the inference code was not choosing the correct model type (it was choosing the basic model instead of the residual one). Pull the changes and it should work. Please let me know if any further issues crop up.

[shikhar-scs]

cool man, thanks its working !
Also, one more thing, I wonder if anyone else faced it too, this line

LipGAN/train.py

Line 71 in 81c5340

img_ip = cv2.imread(os.path.join(dir_name, ip_fname))

should really be just

img_ip = cv2.imread(ip_fname) 

a similar edit for train_unet.py
Just, check it before you make an edit (might be some namespacing error from my side instead). It wasn't working in my case and I got corrupt reads from the line that follows it.

Thanks anyways !