@ -358,7 +358,7 @@ def _decode_predictions(predictions: np.ndarray,
|
|||||||
image_size: int,
|
image_size: int,
|
||||||
entropy_threshold: float,
|
entropy_threshold: float,
|
||||||
confidence_threshold: float,
|
confidence_threshold: float,
|
||||||
iou_thresholdd: float) -> np.ndarray:
|
iou_threshold: float) -> np.ndarray:
|
||||||
return decode_func(
|
return decode_func(
|
||||||
y_pred=predictions,
|
y_pred=predictions,
|
||||||
img_width=image_size,
|
img_width=image_size,
|
||||||
@ -366,7 +366,7 @@ def _decode_predictions(predictions: np.ndarray,
|
|||||||
input_coords="corners",
|
input_coords="corners",
|
||||||
entropy_thresh=entropy_threshold,
|
entropy_thresh=entropy_threshold,
|
||||||
confidence_thresh=confidence_threshold,
|
confidence_thresh=confidence_threshold,
|
||||||
iou_threshold=iou_thresholdd
|
iou_threshold=iou_threshold
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user