Fix plt output comment (#1428)

This commit is contained in:
ian 2023-08-03 23:35:52 -07:00 committed by GitHub
parent 801bed4f66
commit c08ed1949f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -50,7 +50,7 @@ def infer(model, img):
# if you want to look at the outputs
"""
import matplotlib.pyplot as plt
plt.plot(out.numpy()[0])
plt.plot(out[0])
plt.show()
"""
return out, retimg