Skip to content

Precision-Recall Curve Plotting in test.py #1105

@glenn-jocher

Description

@glenn-jocher

🚀 Feature

TODO: PR Curve Autoplot on training finish. See #853 (comment)

yolov5/utils/general.py

Lines 295 to 304 in 9eae82e

# Plot
# fig, ax = plt.subplots(1, 1, figsize=(5, 5))
# ax.plot(recall, precision)
# ax.set_xlabel('Recall')
# ax.set_ylabel('Precision')
# ax.set_xlim(0, 1.01)
# ax.set_ylim(0, 1.01)
# fig.tight_layout()
# fig.savefig('PR_curve.png', dpi=300)

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions