(data)
| 61 | plt.show() |
| 62 | |
| 63 | def plot_val(data): |
| 64 | plt.title(os.path.basename(sys.argv[1]) + ' Validation mAP') |
| 65 | plt.xlabel('Epoch') |
| 66 | plt.ylabel('mAP') |
| 67 | |
| 68 | x = [x[1]['epoch'] for x in data if x[0]['type'] == 'box'] |
| 69 | plt.plot(x, [x[0]['all'] for x in data if x[0]['type'] == 'box']) |
| 70 | plt.plot(x, [x[0]['all'] for x in data if x[0]['type'] == 'mask']) |
| 71 | |
| 72 | plt.legend(['BBox mAP', 'Mask mAP']) |
| 73 | plt.show() |
| 74 | |
| 75 | if len(sys.argv) > 2 and sys.argv[2] == 'val': |
| 76 | plot_val(data['val']) |