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

Object Detection Quantization fails with OpenVINO 2024.3 #2889

Open
1 task
fmishra747 opened this issue Aug 14, 2024 · 2 comments
Open
1 task

Object Detection Quantization fails with OpenVINO 2024.3 #2889

fmishra747 opened this issue Aug 14, 2024 · 2 comments
Labels
bug Something isn't working

Comments

@fmishra747
Copy link

🐛 Describe the bug

The yolov8 object detection quantization fails at
File /lib/python3.10/site-packages/ultralytics/models/yolo/detect/val.py", line 183, in get_stats

**self.nt_per_image = np.bincount(stats["target_img"].astype(int), minlength=self.nc)

KeyError: 'target_img'**
Complete log:

ov2024 (2).3_aa_obj_det.md

Environment

Yolov8 model with coco128 dataset Object Detection Quantization fails with
OpenVINO 2024.3,
"python_version": "3.10",
nncf 2.13.0.dev0+eb91af2d
ultralytics 8.2.74
ultralytics-thop 2.0.0
torch 2.4.0+cpu
torchvision 0.19.0+cpu

The same application works fine with
OpenVINO 2024.1
"python_version": "3.10",
ultralytics 8.2.18
nncf 2.11.0.dev0+24f0ee70
torch 2.2.2+cpu
torchvision 0.17.2+cpu

Minimal Reproducible Example

Execute object detection quantization python script after changing the extension from .md file
quantize_accuracy_aware_object_detection_fullyworking.py.md
with yolov8 and coco128 dataset

Are you going to submit a PR?

  • Yes I'd like to help by submitting a PR!
@fmishra747 fmishra747 added the bug Something isn't working label Aug 14, 2024
@daniil-lyakhov
Copy link
Collaborator

daniil-lyakhov commented Aug 14, 2024

@fmishra747, your error log and the reproducer code does not match, please remove all unused сode and attach the reproducer as a text comment

@MaximProshin
Copy link
Collaborator

@fmishra747 , can you please provide the input requested above? Is still valid?

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

3 participants