將caffe訓練時loss的變化曲線用matlab繪制出來


1. 首先是提取 訓練日志文件;

 

2. 然后是matlab代碼:

clear all; close all; clc; log_file = '/home/wangxiao/Downloads/43_attribute_baseline.log'; fid = fopen(log_file, 'r'); fid_accuracy = fopen('/home/wangxiao/Downloads/output_accuracy.txt', 'w'); fid_loss = fopen('/home/wangxiao/Downloads/output_loss.txt', 'w'); iteration ={}; loss = {}; accuracy = {}; path = '/home/wangxiao/Downloads/'; fid_ = fopen([path, 'loss_file_.txt'], 'a'); while(~feof(fid)) tline = fgetl(fid); %% if strfind(tline, 'sgd_solver.cpp:') iter_index = strfind(tline, 'Iteration '); rest = tline((iter_index+9):end); iter_current = strtok(rest, ','); % iteration number; iteration = [iteration iter_current]; % count the iteration; lr_index = strfind(tline, 'lr = '); lr_current = tline((lr_index+4):end); % learning rate; end %% if strfind(tline, 'solver.cpp:228]') iter_index = strfind(tline, 'loss = '); loss_current = tline((iter_index+7):end); fprintf(fid_, '%s \n', loss_current ); loss = [loss loss_current] ; % count the iteration; end if strfind(tline, 'aver_accuracy: ') aver_accuracy_index = strfind(tline, 'aver_accuracy: '); aver_accuracy_current = tline((aver_accuracy_index+15):end); accuracy = [accuracy aver_accuracy_current]; end end loss_file_Path = importdata('/home/wangxiao/Downloads/loss_file_.txt'); plot(loss_file_Path)

 

3. 結果展示:


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM