-
Notifications
You must be signed in to change notification settings - Fork 16
/
prepare_images.m
71 lines (60 loc) · 2.27 KB
/
prepare_images.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
function prepare_images(opts,data)
disp('* Preparing images *');
if ~exist(opts.fileImages, 'file')
% Prepare header
% Get bytes per image after resizing
bytesPerImage = int64(zeros(1,length(data.words)));
for i=1:length(data.words)
H = data.words(i).H;
W = data.words(i).W;
if (opts.minH > data.words(i).H)
W = ceil((opts.minH * data.words(i).W)/data.words(i).H-1e-5);
H = opts.minH;
end
if (opts.maxH < data.words(i).H)
W = ceil((opts.maxH * data.words(i).W)/data.words(i).H-1e-5);
H = opts.maxH;
end
bytesPerImage(i) = 8 + W*H;% Two bytes for image header (rows, cols) plus k elements as uint8
end
posAtI = 4 + length(bytesPerImage)* 8 + [ 0 cumsum(double(bytesPerImage))]; % Numer of images + lookup (8 byte integers) + accumulated sizes
posAtI = int64(posAtI(1:end-1));
fid = fopen(opts.fileImages, 'w');
fwrite(fid, int32(length(bytesPerImage)), 'int32');
fwrite(fid, posAtI, 'int64');
previousPath = '';
for i=1:length(data.words)
if ~strcmp(previousPath,data.words(i).pathIm)
previousPath = data.words(i).pathIm;
im = imread(data.words(i).pathIm);
[Him,Wim,numC] = size(im);
if numC > 1
im = rgb2gray(im);
end
end
patch = im(data.words(i).loc(3):data.words(i).loc(4),data.words(i).loc(1):data.words(i).loc(2));
% Move to single and equalize if necessary
patch = im2single(patch);
m = max(max(patch));
if m < 0.2
patch = patch*0.2/m;
end
[H,W] = size(patch);
if (data.words(i).H~=H || data.words(i).W~=W)
error('something wrong happened!');
end
if (opts.minH > data.words(i).H)
patch = imresize(patch, [opts.minH,nan]);
end
if (opts.maxH < data.words(i).H)
patch = imresize(patch, [opts.maxH,nan]);
end
[H,W,numC] = size(patch);
% Save as uint8
fwrite(fid, int32(W),'int32');
fwrite(fid, int32(H), 'int32');
fwrite(fid, im2uint8(patch), 'uint8');
end
fclose(fid);
end
end