search

bit-bots / YOEO

YouOnlyEncodeOnce - A CNN for Embedded Object Detection and Semantic Segmentation
GNU General Public License v3.0
22 stars 4 forks source link

Feature/convert to onnx and openvino scripts #39

Closed phinik closed 1 year ago

phinik commented 2 years ago

2 Scripts to convert a PyTorch model to ONNX and from ONNX to OpenVino IR.

TODO:

Flova commented 2 years ago

Ping @phinik ? The scripts lgtm maybe we should also open a upstream pr at pytorchyolo.

jaagut commented 2 years ago

@Flova Have you tested the scripts? I believe this is ready for review/merge.

jaagut commented 2 years ago

We can ignore the failing CI for now, as it is not yet fully working...

phinik commented 2 years ago

@Flova Have you tested the scripts? I believe this is ready for review/merge.

pls, don't merge yet.

phinik commented 2 years ago

@Flova Have you tested the scripts? I believe this is ready for review/merge.

pls, don't merge yet.

Can be merged

phinik commented 2 years ago

Ping @phinik ? The scripts lgtm maybe we should also open a upstream pr at pytorchyolo.

Pytorchyolo doesn't have scripts yet, right?

Flova commented 2 years ago

That is correct. I would suggest adding it.

Flova commented 2 years ago

We can also add them to the commands defined in the pyproject,toml. So they are available if it is installed via pip without direct code access.