search

FZJ-INM1-BDA / celldetection

Scalable Instance Segmentation using PyTorch & PyTorch Lightning.
https://docs.celldetection.org
Apache License 2.0
125 stars 21 forks source link

convert to onnx inference #5

Open fanweiya opened 2 years ago

fanweiya commented 2 years ago

Can you give some onnx inference example code? I cant find how to get the convert ONNX file and post processing? Many thanks !

ericup commented 2 years ago

Very good point! With the current version that's a bit tricky and would only work with monkey patching. I'll try to add support and documentation for this as soon as possible.

Reeeeev commented 4 months ago

The same issue occurred. I tried converting the model to ONNX for inference, but it failed.