Shift-Net_pytorch
Shift-Net_pytorch copied to clipboard
About the mask
Can this model trained on other mask dataset?
Hello, It can be trained on any dataset You just need to put --dataroot to ur data, and maybe change the dataLoader to use the correct path.
Best,