-
Notifications
You must be signed in to change notification settings - Fork 432
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
burn-import: ConvTranspose support #883
Comments
@npatsakula, it has been successfully merged. I recommend looking into my SqueezeNet ONNX conversation model: SqueezeNet-Burn. Note that SqueezeNet is primarily an image classification model. This version has examples of how to feed an image and utilizes an image normalizer, similar to what onnx-ocr employs. I'm eager to see what you develop. If you're interested, we could host the model here: Tracel-AI Models. The ONNX file could be downloaded during a build process (using build.rs). I'd be happy to contribute as well. CC: @nathanielsimard, @Luni-4 |
@antimora, thank you a lot! I used your SqueezeNet wrapper as a source of inspiration. Unfortunately I'm a little stuck with inference results post-processing (onnx-ocr makes heavy use of PaddleOCR internals; I just don't have time to look at it properly). I plan to to get to working on this after the New Year. I'm planning to use onnx-ocr burn wrapper for two purposes:
|
Linking to the issue to port into burn/models: tracel-ai/models#23 |
Feature description
ConvTranspose support for
burn-import
ONNX converter.Feature motivation
I've tried to port onnx-ocr to
burn-rs
usingburn-import
and stuck without support ofConvTranspose
(two-dimensional version).Suggest a Solution
I tried to add support by myself and stuck with proper support of constant weights (see my PR).
The text was updated successfully, but these errors were encountered: