1. 程式人生 > >Detectron繪製loss曲線和pr曲線

Detectron繪製loss曲線和pr曲線

繪製loss曲線

參見部落格:detectron訓練日誌視覺化

繪製p-r曲線

在detectron測試完畢後,會在相應的結果輸出資料夾中輸出每一類的不同precision對應的recall值,儲存在xx_pr.pkl中,例如ship_pr.pkl,繪製p-r的程式碼為:

import pickle
import os.path as osp
import matplotlib.pyplot as plt

PR_path= ''
f = open(PR_path,'rb')  
info = pickle.load(f)
precision = info['prec']
recall = info['rec']
plt.plot(recall,precision)
plt.show()