Skip to content

Commit ea11cf8

Browse files
authored
Fix inference YOLO model on grayscale images (#9267)
1 parent 208b4ed commit ea11cf8

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed
Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
### Fixed
2+
3+
- Fixed inference with the YOLOv7 model on grayscale images
4+
(<https://github.com/cvat-ai/cvat/pull/9267>)

serverless/onnx/WongKinYiu/yolov7/nuclio/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ def handler(context, event):
2929
data = event.body
3030
buf = io.BytesIO(base64.b64decode(data["image"]))
3131
threshold = float(data.get("threshold", 0.5))
32-
image = Image.open(buf)
32+
image = Image.open(buf).convert("RGB")
3333

3434
results = context.user_data.model.infer(image, threshold)
3535

0 commit comments

Comments
 (0)