search

layer6ai-labs / GSS

Code for the NeurIPS'19 paper "Guided Similarity Separation for Image Retrieval"
64 stars 7 forks source link

pytorch version #1

Open IssamLaradji opened 4 years ago

IssamLaradji commented 4 years ago

Great work!

Will a pytorch version be released for this code?

Thanks!

chundiliu commented 4 years ago

Hi Issam,

Thanks for being interested in our work! We do have a pytorch implementation, but it's not in shape yet. We will let you know once we clean everything up. But if you look into the code, most of the complicated operations are preprocessing and they are written in numpy. It won't be too hard to reuse them and rewrite the pytorch training part by yourself. Hope it helps.

Chundi