Skip to content

Commit

Permalink
Update attention_simulation_contri_plots.m
Browse files Browse the repository at this point in the history
  • Loading branch information
huangpei90 authored Jul 15, 2020
1 parent 328c1e7 commit 44fbbf3
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions attention_simulation_contri_plots.m
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
function attention_simulation_contri_plots(res)
function attention_simulation_contri_plots(res_file)
% Plot results from the output of attention_simulation.m
% This function queries the output for the parameters we are interested in
% and generate plots comparing the ground truth against the various
Expand All @@ -7,11 +7,10 @@ function attention_simulation_contri_plots(res)
% If the input res is undefined, we load the sample result from
% sample_results/att_sim_results.mat
%
% attention_simulation_contri_plots(res)
% attention_simulation_contri_plots(res_file)

if ~exist('res', 'var') || isempty(res)
res = load(fullfile(fileparts(mfilename('fullpath')), ...
'sample_results', 'att_sim_results.mat'));
if ~exist('res_file', 'var') || isempty(res_file)
res_file = fullfile(fileparts(mfilename('fullpath')),'sample_results', 'att_sim_results.mat');
end

% define which values we are interested in
Expand All @@ -25,6 +24,7 @@ function attention_simulation_contri_plots(res)
179,205,227];
cmap = cmap/255;

res = load(res_file);
res = res.results;

%Find the indices for the 3 rows
Expand Down

0 comments on commit 44fbbf3

Please sign in to comment.