-
Notifications
You must be signed in to change notification settings - Fork 148
Open
Description
Hello, I wonder if my calculation of the final f1 in this way is correct?
Under what conditions were the precision and recall returned by this model obtained?
I want the f1 of the detection box at iou=0.5.
validator_metrics = model.val(
data=f'{args.data}.yaml',
batch=args.batch_size,
device=args.device,
)
result = validator_metrics.results_dict
precision = result['metrics/precision(B)']
recall = result['metrics/recall(B)']
F1 = 2 * precision * recall / (precision + recall)
Metadata
Metadata
Assignees
Labels
No labels