Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

Precision-Recall Curve Plotting in test.py #1105

Closed
glenn-jocher opened this issue Oct 9, 2020 · 0 comments 路 Fixed by #1107
Closed

Precision-Recall Curve Plotting in test.py #1105

glenn-jocher opened this issue Oct 9, 2020 · 0 comments 路 Fixed by #1107
Assignees
Labels
enhancement New feature or request

Comments

@glenn-jocher
Copy link
Member

glenn-jocher commented Oct 9, 2020

馃殌 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)

@glenn-jocher glenn-jocher added enhancement New feature or request TODO labels Oct 9, 2020
@glenn-jocher glenn-jocher self-assigned this Oct 9, 2020
@glenn-jocher glenn-jocher linked a pull request Oct 9, 2020 that will close this issue
@glenn-jocher glenn-jocher removed the TODO label Oct 11, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant