File size: 730 Bytes
63ef347 | 1 2 3 4 5 6 7 | from pathlib import Path
import sys,numpy as np;import matplotlib;matplotlib.use('Agg');import matplotlib.pyplot as plt
R=Path(__file__).resolve().parents[1];sys.path.insert(0,str(R));from model.climatenet import cfg,write
c=cfg(R);d=np.load(R/c['paths']['predictions']);p=d['prediction'];t=d['target'];ious=[]
for k in range(3):inter=((p==k)&(t==k)).sum();union=((p==k)|(t==k)).sum();ious.append(float(inter/max(union,1)))
write(R/c['paths']['evaluation'],{'class_iou':ious,'mean_iou':float(np.mean(ious)),'is_complete_global':False,'synthetic':True});fig,ax=plt.subplots(1,2,figsize=(9,4));ax[0].imshow(t[0]);ax[1].imshow(p[0]);q=R/c['paths']['figure'];q.parent.mkdir(parents=True,exist_ok=True);plt.savefig(q,dpi=150);print(q)
|