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