search

bytecodealliance / wasi-nn

High-level bindings for wasi-nn system calls
Apache License 2.0
41 stars 33 forks source link

image2tensor supports planar format output. #92

Closed jianjunz closed 7 months ago

jianjunz commented 7 months ago

This change mainly adds support for outputting tensor bytes in planar memory layout, or CHW format. Besides this, it also adds support for png input with alpha channel (output still doesn't have alpha channel, so it's stripped out).

abrown commented 7 months ago

cc: @geekbeast

abrown commented 7 months ago

I guess one of us should try to fix CI; @jianjunz, we can talk about this more next week. The current job is a bit of a scripted monolith that we might need to refactor. But then again, it's not too useful to spend a lot of time on this since the bindings portion of this could be replaced by wit-bindgen-generated bindings for wasi-nn.

jianjunz commented 7 months ago

Thanks for your review. Please help merge if it looks good.

The CI failure is caused by openvino checksum mismatch. It may need an update.

geekbeast commented 7 months ago

A bit late, but this looks good to me.