(self, filename, pred, label, outdir)
| 119 | return image_dir, input_file |
| 120 | |
| 121 | def paint(self, filename, pred, label, outdir): |
| 122 | |
| 123 | num = len(filename) |
| 124 | |
| 125 | for idx in range(num): |
| 126 | ax, fig, h, w = self.get_axis(filename[idx]) |
| 127 | self.paint_one_image(pred[idx], label[idx], ax, h, w) |
| 128 | out_name = os.path.join(outdir, os.path.basename(filename[idx])) |
| 129 | fig.savefig(out_name, dpi=200) |
| 130 | plt.close('all') |
| 131 | |
| 132 | @staticmethod |
| 133 | def paint_one_image(pred, label, ax, h, w): |
no test coverage detected