python画loss随迭代次数的图_YOLOv3之loss和iou可视化（横坐标和纵坐标与迭代次数完美对齐）...

建议大家先去看看其他博客的代码，就能体会到它的魅力了。

Table of Contents

一、extract_log.py

二、visualization_loss.py

三、visualization_iou.py

一、extract_log.py

#!/usr/bin/python

#coding=utf-8

#该文件用于提取训练log，去除不可解析的log后使log文件格式化，生成新的log文件供可视化工具绘图

import inspect

import os

import random

import sys

def extract_log(log_file, new_log_file, key_word):

with open(log_file, 'r') as f:

with open(new_log_file, 'w') as train_log:

for line in f:

#去除多GPU的同步log；去除除零错误的log

if ('Syncing' in line) or ('nan' in line):

continue

if key_word in line:

train_log.write(line)

f.close()

train_log.close()

extract_log('./2048/train_log2.txt', './2048/log_loss2.txt', 'images')

extract_log('./2048/train_l