search

Relja / netvlad

NetVLAD: CNN architecture for weakly supervised place recognition
MIT License
534 stars 122 forks source link

Is it possible to convert to ONNX format? #55

Closed fettahyildizz closed 1 year ago

fettahyildizz commented 1 year ago

Is it possible to convert to ONNX format? We want to work with TensorRT environment for runtime purpose.

Relja commented 1 year ago

Hi, Sorry I don't know about this as I never used pytorch and haven't been working on NetVLAD-related stuff in 7 years. Let me point you to a previous thread: https://github.com/Relja/netvlad/issues/43 Also there are some pytorch reimplementations of NetVLAD, e.g. you could try this: https://github.com/Nanne/pytorch-NetVlad