How to set numItemsThreshold #149
Description
Hi there.
I'm declaring my model as YOLO like this:
private func getDetectionModel() async -> YOLO {
let model: YOLO = await withCheckedContinuation { continuation in
_ = YOLO("pvp_weights", task: .detect) { result in
switch result {
case .success(let initializedModel):
print("detector initialized")
continuation.resume(returning: initializedModel)
case .failure(let error):
continuation.resume(throwing: error as! Never)
}
}
}
return model
}
There is no way to set it's numItemsThreshold. It's set to 30 by default.
setNumItemsThreshold() function is declared in BasePredictor but it's not accessible through YOLO model type. Or we can not set it using it's initializer.
Is there any way to increase this variable?