search

rdroste / unisal

Unified Image and Video Saliency Modeling (ECCV 2020)
https://arxiv.org/abs/2003.05477
Apache License 2.0
128 stars 36 forks source link

unisal can't transfer onnx model #15

Closed upupbo closed 2 years ago

upupbo commented 2 years ago

Dear author, your work is very good, but your network is operated with torch.unbind everywhere, which makes it impossible to export the onnx model and onnx acceleration, so sad, have you tried this, thank you

rdroste commented 2 years ago

Sorry, we didn't design the network to work with ONNX. Feel free to fork and adapt.