Skip to content
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

ezkl cli gen-settings: [graph] a node is missing required params: padding #840

Open
pdibitonto opened this issue Aug 28, 2024 · 1 comment
Labels
bug Something isn't working

Comments

@pdibitonto
Copy link

I was testing ezkl with a publicly available model YOLOv8s-worldv2 but I'm receiving an error when I fire the gen-settings command, the error is the following:
[graph] a node is missing required params: padding

Steps to reproduce the bug

To reproduce the bug:

  1. Download the YOLOv8s-worldv2 model here

  2. Export the model to onnx with the following python script:

from ultralytics import YOLO
model = YOLO("yolov8s-worldv2.pt")
model.export(format = "onnx")
  1. In the same directory, try the gen-settings cli command as follow:
ezkl gen-settings --model yolov8s-worldv2.onnx --input-visibility hashed --output-visibility public

Device and Operating System

OS: Ubuntu 22.04.4
Version: 22.04.4 LTS
Processor: Intel® Core™ i7-10750H
Memory: 32 GB

Additional Information

  1. Python version: 3.8.16+
  2. Required python module for onnx export: ultralytics
@pdibitonto pdibitonto added the bug Something isn't working label Aug 28, 2024
@pdibitonto
Copy link
Author

Dunno if those details are useful, but I've printed the model (already exported to onnx) graph padding values and they are presents for each node, I've used this script:

import onnx

# Load the ONNX model
model = onnx.load("yolov8s-worldv2.onnx")

# Access the graph
graph = model.graph


for node in graph.node:
    if node.op_type == 'Conv':  # Look for Convolution nodes
        print(f"Node name: {node.name}")
        for attr in node.attribute:
            if attr.name == 'pads':  # 'pads' attribute contains padding info
                print(f"Padding: {attr.ints}")

Let me know if others infos are needed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

1 participant