Skip to content

Commit 0241714

Browse files
authored
Update eval_all.m
1 parent 13096c0 commit 0241714

File tree

1 file changed

+7
-12
lines changed

1 file changed

+7
-12
lines changed

evaluation/eval_all.m

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,10 @@
1212
switch data_name
1313
case 'ADE20K'
1414
isVal = true; %evaluation on valset
15-
step = 2000; %equals to number of images divide num of GPUs in testing e.g. 500=2000/4
15+
step = 500; %equals to number of images divide num of GPUs in testing e.g. 500=2000/4
1616
data_root = '/data2/hszhao/dataset/ADEChallengeData2016'; %root path of dataset
1717
eval_list = 'list/ADE20K_val.txt'; %evaluation list, refer to lists in folder 'samplelist'
18-
eval_list = 'validation.txt'; %evaluation list, refer to lists in folder 'samplelist'
19-
save_root = 'mc_result/ADE20K/val/pspnet50_473_g0/'; %root path to store the result image
18+
save_root = 'mc_result/ADE20K/val/pspnet50_473/'; %root path to store the result image
2019
model_weights = 'model/pspnet50_ADE20K.caffemodel';
2120
model_deploy = 'prototxt/pspnet50_ADE20K_473.prototxt';
2221
fea_cha = 150; %number of classes
@@ -26,10 +25,9 @@
2625
data_colormap = 'color150.mat'; %color map
2726
case 'VOC2012'
2827
isVal = false; %evaluation on testset
29-
step = 364;
28+
step = 364; %364=1456/4
3029
data_root = '/data2/hszhao/dataset/VOC2012';
3130
eval_list = 'list/VOC2012_test.txt';
32-
eval_list = 'list/test.txt';
3331
save_root = 'mc_result/VOC2012/test/pspnet101_473/';
3432
model_weights = 'model/pspnet101_VOC2012.caffemodel';
3533
model_deploy = 'prototxt/pspnet101_VOC2012_473.prototxt';
@@ -40,10 +38,9 @@
4038
data_colormap = 'colormapvoc.mat';
4139
case 'cityscapes'
4240
isVal = true;
43-
step = 125;
41+
step = 125; %125=500/4
4442
data_root = '/data2/hszhao/dataset/cityscapes';
4543
eval_list = 'list/cityscapes_val.txt';
46-
eval_list = 'list/fine_val.txt';
4744
save_root = 'mc_result/cityscapes/val/pspnet101_713/';
4845
model_weights = 'model/pspnet101_cityscapes.caffemodel';
4946
model_deploy = 'prototxt/pspnet101_cityscapes_713.prototxt';
@@ -55,25 +52,23 @@
5552
end
5653
skipsize = 0; %skip serveal images in the list
5754

58-
is_save_feat = false; %set to true if final feature map is needed
55+
is_save_feat = false; %set to true if final feature map is needed (not suggested for storage consuming)
5956
save_gray_folder = [save_root 'gray/']; %path for predicted gray image
6057
save_color_folder = [save_root 'color/']; %path for predicted color image
6158
save_feat_folder = [save_root 'feat/']; %path for predicted feature map
6259
scale_array = [1]; %set to [0.5 0.75 1 1.25 1.5 1.75] for multi-scale testing
63-
%scale_array = [0.5 0.75 1 1.25 1.5 1.75];
6460
mean_r = 123.68; %means to be subtracted and the given values are used in our training stage
6561
mean_g = 116.779;
6662
mean_b = 103.939;
6763

6864
acc = double.empty;
6965
iou = double.empty;
70-
gpu_id_array = [12:15]; %multi-GPUs for parfor testing, if number of GPUs is changed, remember to change the variable 'step'
71-
gpu_id_array = [1]; %multi-GPUs for parfor testing, if number of GPUs is changed, remember to change the variable 'step'
66+
gpu_id_array = [0:3]; %multi-GPUs for parfor testing, if number of GPUs is changed, remember to change the variable 'step'
7267
runID = 1;
7368
gpu_num = size(gpu_id_array,2);
7469
index_array = [(runID-1)*gpu_num+1:runID*gpu_num];
7570

76-
for i = 1:gpu_num %change 'parfor' to 'for' if singe GPU testing is used
71+
parfor i = 1:gpu_num %change 'parfor' to 'for' if singe GPU testing is used
7772
eval_sub(data_name,data_root,eval_list,model_weights,model_deploy,fea_cha,base_size,crop_size,data_class,data_colormap, ...
7873
is_save_feat,save_gray_folder,save_color_folder,save_feat_folder,gpu_id_array(i),index_array(i),step,skipsize,scale_array,mean_r,mean_g,mean_b);
7974
end

0 commit comments

Comments
 (0)