search

rmitra / PS-Dataset

PhotoSynth Dataset for improving local patch Descriptors
74 stars 6 forks source link

HardNet training #4

Open amitshaked opened 6 years ago

amitshaked commented 6 years ago

Hey,

Can you please upload your pytorch code for processing the dataset and use it to train the network with HardNet loss?

Thanks!

Shu-HowTing commented 5 years ago

+1

rmitra commented 5 years ago

We used Torch7 framework in our implementation. I will recommend to refer to https://github.com/DagnyT/hardnet for the most efficient pytorch implementation.