search

scaelles / DEXTR-PyTorch

Deep Extreme Cut http://www.vision.ee.ethz.ch/~cvlsegmentation/dextr
GNU General Public License v3.0
843 stars 153 forks source link

Model conversion #16

Closed bsekachev closed 5 years ago

bsekachev commented 5 years ago

As far as I understand you have converted PyTorch weights to Keras weighs. Right? How have you done it? Have you used an intermediate ONNX format?

scaelles commented 5 years ago

Hello, I didn't use ONNX, I manually saved all the weights into numpy and then load them from numpy in Keras. You can find the script that I used here and then the function to load from numpy here

Let me know if you have any further question.

bsekachev commented 5 years ago

Hello, Thank you so much.

bhavesh907 commented 4 years ago

Hi, @scaelles how did you convert your PyTorch model in .bin and .xml format for cvat tool? .