Skip to content

Commit 7fcea67

Browse files
committed
Fixed BGR to RGB conversion.
1 parent e110918 commit 7fcea67

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

inference/app.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def get_model(model_ref: str):
4545
def run(model_ref, image, conf):
4646
model = get_model(model_ref)
4747
results = model(image, conf=conf)
48-
annotated = results[0].plot() # BGR -> RGB
48+
annotated = results[0].plot()[..., ::-1] # BGR -> RGB
4949

5050
names = model.names
5151

0 commit comments

Comments
 (0)