darknet - 如何修改 YOLOv4 的损失图表？

Question

我正在使用来自这个 Github 存储库的 YOLOv4：https ://github.com/AlexeyAB/darknet

从一个教程（见这里：Github theAIGuysCode）我看到我可以用这段代码绘制损失图：

def imageViewer(path):
  import cv2
  import matplotlib.pyplot as plt
  %matplotlib inline
  image = cv2.imread(path)
  fig = plt.gcf()
  fig.set_size_inches(12, 8)
  plt.imshow(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
  plt.show()

问题：

在前 800 次迭代中，我的网络的平均损失约为 80。不过我想在图表中看到它。该图表仅显示 [0, 20] 范围内的损失值，因为这些是更常见的值。我尝试使用简单的 matplotlib 函数（如 .ylim() 等）修改图表，但我注意到图表本身无法使用 matplotlib 修改。您知道是否有可能修改此图表以便我可以更改 y 轴的范围？

谢谢！！！