Skip to content

Commit 519503e

Browse files
author
Kasper
committed
Fix inference crashing when extra_args is left None
1 parent 60fa754 commit 519503e

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

patched_yolo_infer/elements/CropElement.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ def __init__(
3535
def calculate_inference(self, model, imgsz=640, conf=0.35, iou=0.7, segment=False, classes_list=None, memory_optimize=False, extra_args=None):
3636

3737
# Perform inference
38+
extra_args = {} if extra_args is None else extra_args
3839
predictions = model.predict(self.crop, imgsz=imgsz, conf=conf, iou=iou, classes=classes_list, verbose=False, **extra_args)
3940

4041
pred = predictions[0]

0 commit comments

Comments
 (0)