DVBPR icon indicating copy to clipboard operation
DVBPR copied to clipboard

How to train on custom data

Open ayush9198gupta opened this issue 4 years ago • 1 comments

Hi,

@kang205 I am using pretrained data of AmazonFashion (Fashion6ImgPartitioned.npy) with your model and it is working but I want to extend your model with custom data more specifically on full body images (custom data).

Points to be focus on:-

  1. Can model will work for full body images for outfits recommendation
  2. Can we do transfer learning on custom data/pretrained data.

Kindly suggest Thanks

ayush9198gupta avatar Apr 30 '20 08:04 ayush9198gupta

I think the recommendation part could be extended to full-body images, and you should make some slight modifications on the code to do the transfer or adaptation. However, for the generation part, It might be hard since GAN didn't handle full-body very well till today.

kang205 avatar May 01 '20 23:05 kang205