kang205 / DVBPR

Personalized Fashion Recommendation and Generation
188 stars 51 forks source link

How to train on custom data #5

Open ayush9198gupta opened 4 years ago

ayush9198gupta commented 4 years ago

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

kang205 commented 4 years ago

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.