Giters

deeplearning4j / deeplearning4j

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learning using automatic differentiation.

Home Page:http://deeplearning4j.konduit.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can't import yolov8

huzpsb opened this issue · comments

commented

Unable to resolve attribute for name auto_pad for node Conv for op type Conv
Unable to resolve attribute for name dilations for node MaxPool for op type MaxPool
Skipping input B on node /model.22/dfl/conv/Conv

These console logs exist when I import yolov8x.onnx, is it normal and can be ignore? Thanks.

model link:
https://huggingface.co/spaces/Booker8/ll/resolve/main/yolov8x.zip

commented

@huzpsb I'll need to check with the latest version. There's quite a bit of work that went in to the newest release + model import. I believe we should be ok here though. Thanks for filing!

commented

I'm facing the same problem. Are there any solutions? Or another way to load Yolov8?

commented

I'm facing the same problem. Are there any solutions? Or another way to load Yolov8?

Try Python + RESTful apis. :/
A quick work-around

commented

@huzpsb @fizlrock I need to publish a release. Part of that will be a CLI tool for conversion. If you want to run python and control it you can try the python4j module which can help with that.