Shift-Net_pytorch icon indicating copy to clipboard operation
Shift-Net_pytorch copied to clipboard

Published 20 hours ago • verified publisher icon Zhaoyi-Yan

Any help on custom dataset?

Open tiwarikaran opened this issue 3 years ago • 4 comments

Hello, I am absolutely struggling with how, to begin with, image inpainting and this seems like a good algorithm to work on, any help whatsoever would be appreciated. Also, I tried to run the code but got errors.

tiwarikaran avatar Mar 21 '21 15:03 tiwarikaran

You can post the error information

Zhaoyi-Yan avatar Mar 21 '21 15:03 Zhaoyi-Yan

Before that, how should I prepare my dataset?

tiwarikaran avatar Mar 23 '21 13:03 tiwarikaran

Put all your training images to the custom path, then change dataroot to your custom path in https://github.com/Zhaoyi-Yan/Shift-Net_pytorch/blob/master/options/base_options.py.

I have updated Readme here: https://github.com/Zhaoyi-Yan/Shift-Net_pytorch#train-models

Zhaoyi-Yan avatar Mar 24 '21 02:03 Zhaoyi-Yan

Is it necessary for the training images to have mask or the mast is randomly generated?

tiwarikaran avatar Mar 24 '21 18:03 tiwarikaran