조현아

data cleaning for each class

1 +
2 +% 맨처음 방향 (frame1) 으로 일괄 orientation normalize
3 +% => 1번프레임이 아니라 프레임들의 중앙값
4 +% 키로 모든방향 normalize
5 +% 노멀라이즈시 최소값 빼는게 아니라 joint1-> 0.5/0.5/0.5로 가도록
6 +% 실험할때 전방향 노멀라이즈도 해봐야겠다..
7 +path_name ='/home/hyuna/Documents/actionGAN_work/20_cleansed_skeleton/';
8 +dinfo=dir('/home/hyuna/Documents/actionGAN_work/20_cleansed_skeleton/*.skeleton');
9 +%txt=fullfile(dir_to_search, '*.skeleton');
10 +%dinfo=dir(txt);
11 +count=[0,0,0];
12 +for d = 1: length(dinfo)
13 + label=-1;
14 + file_name = dinfo(d).name;
15 + % disp(name);
16 +
17 + name = strcat(path_name,file_name(1:20),'.skeleton');
18 + [token,remainder] = strtok(file_name,'A');
19 +
20 + class = str2num(remainder(2:4));
21 + %class=remainder(2:4);
22 +
23 + if class == 20
24 + bodyinfo = read_skeleton_file(name);
25 + frame_num = size(bodyinfo,2);
26 +
27 + else
28 + continue;
29 + end
30 +
31 + if isempty(bodyinfo) || isempty(bodyinfo(1).bodies())
32 + disp("empty body");
33 + newdir='/home/hyuna/Documents/actionGAN_work/20_emptybody.txt';
34 +
35 + txtfile=fopen(newdir,'a');
36 + fprintf(txtfile, file_name(1:20));
37 + fprintf(txtfile, '\n');
38 + fclose(txtfile);
39 + count(1)=count(1)+1;
40 + continue;
41 + end
42 +
43 +
44 + %initialize
45 + cur_subject_x = zeros(frame_num, 25);
46 + cur_subject_y = zeros(frame_num, 25);
47 + cur_subject_z = zeros(frame_num, 25);
48 +
49 + tot_x = zeros(frame_num,25);
50 + tot_y = zeros(frame_num,25);
51 + tot_z = zeros(frame_num,25);
52 +
53 + joint_5 = zeros(1,3);
54 + joint_9 = zeros(1,3);
55 + joint_1 = zeros(1,3);
56 + joint_3 = zeros(1,3);
57 +
58 +
59 +
60 + %get total joints information
61 + for FN = 1:frame_num
62 +
63 + cur_body = bodyinfo(FN).bodies(1);
64 + joints = cur_body.joints;
65 +
66 + for JN = 1:25
67 + tot_x(FN,JN) = joints(JN).x;
68 + tot_y(FN,JN) = joints(JN).y;
69 + tot_z(FN,JN) = joints(JN).z;
70 + end
71 + end
72 +
73 + %Orientation normalization 1 : in space
74 + %get median values
75 + M_x = median(tot_x);
76 + M_y = median(tot_y);
77 + M_z = median(tot_z);
78 +
79 + %set 3 points for make plane
80 + joint_5 = [M_x(5) M_y(5) M_z(5)];
81 + joint_9 = [M_x(9) M_y(9) M_z(9)];
82 + joint_1 = [M_x(1) M_y(1) M_z(1)];
83 + joint_3 = [M_x(3) M_y(3) M_z(3)];
84 +
85 + %find RIGID TRNASFORMATION matrix
86 + d1 = joint_1 - joint_5;
87 + d2 = joint_1 - joint_9;
88 + n1 = cross(d1,d2); % because we will parallel transform, don't need to find belly
89 + u1 = n1/norm(n1);
90 + u2 = [0 0 1];
91 + cs1 = dot(u1,u2)/norm(u1)*norm(u2);
92 + ss1 = sqrt(1-cs1.^2);
93 + v1 = cross(u1,u2)/norm(cross(u1,u2));
94 +
95 + R1 = [v1(1)*v1(1)*(1-cs1)+cs1 v1(1)*v1(2)*(1-cs1)-v1(3)*ss1 v1(1)*v1(3)*(1-cs1)+v1(2)*ss1];
96 + R1(2,:) = [v1(1)*v1(2)*(1-cs1)+v1(3)*ss1 v1(2)*v1(2)*(1-cs1)+cs1 v1(2)*v1(3)*(1-cs1)-v1(1)*ss1];
97 + R1(3,:) = [v1(1)*v1(3)*(1-cs1)-v1(2)*ss1 v1(2)*v1(3)*(1-cs1)+v1(1)*ss1 v1(3)*v1(3)*(1-cs1)+cs1];
98 +
99 + %1-3 number tolls to parallel x axis. Rigid transformation on plane surface
100 + %Z axis coords oyler angle transform
101 +
102 + t = joint_3 - joint_1;
103 + d3 = R1(1,:) * t.';
104 + d3(1,2) = R1(2,:) * t.';
105 + d3(1,3) = R1(3,:) * t.';
106 +
107 + u3 = d3(1:2)/norm(d3(1:2));
108 + v3 = [u3(1) -u3(2)];
109 + v3(2,:) = [u3(2) u3(1)];
110 + u4 = [1 0].';
111 +
112 + csss = v3\u4;
113 + cs2 = csss(1);
114 + ss2 = csss(2);
115 +
116 + R2 = [cs2 -ss2 0];
117 + R2(2,:) = [ss2 cs2 0];
118 + R2(3,:) = [0 0 1];
119 +
120 +
121 + %apply rigid transformation
122 + for FN = 1:frame_num
123 + cur_body = bodyinfo(FN).bodies(1);
124 + joints = cur_body.joints;
125 +
126 + for JN = 1:25
127 + a = R1(1,:) * [joints(JN).x joints(JN).y joints(JN).z].';
128 + b = R1(2,:) * [joints(JN).x joints(JN).y joints(JN).z].';
129 + c = R1(3,:) * [joints(JN).x joints(JN).y joints(JN).z].';
130 +
131 + cur_subject_x(FN,JN) = R2(1,:) * [a b c].';
132 + cur_subject_y(FN,JN) = R2(2,:) * [a b c].';
133 + cur_subject_z(FN,JN) = R2(3,:) * [a b c].';
134 +
135 + end
136 + end
137 +
138 + %orientation normalize 2 (with plane surface)
139 + if cur_subject_x(1,4) < cur_subject_x(1,1)
140 + cur_subject_x = 0 - cur_subject_x;
141 + end
142 +
143 + if cur_subject_y(1,9) > cur_subject_y(1,5)
144 + cur_subject_y = 0 - cur_subject_y;
145 + end
146 +
147 + % for save origin subjects before data augment
148 + clear_subject_x = cur_subject_x;
149 + clear_subject_y = cur_subject_y;
150 + clear_subject_z = cur_subject_z;
151 +
152 + % patch_num*25
153 + red=[4,3,21,2,1];
154 + green=[5,6,7,8,23,22];
155 + blue=[9,10,11,12,25,24];
156 + skyblue=[17,18,19,20];
157 + yellow=[13,14,15,16];
158 +
159 + target = {skyblue,yellow,green,blue};
160 + [r,c]=size(target);
161 +
162 + %celldisp(target);
163 + %disp(c);
164 + %disp(target{1});
165 +
166 + % joint-to-joint
167 + for tar_index = 1:c
168 + diff_ave = comp_aver(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target{tar_index});
169 +
170 + if label~=0
171 + label=set_label(diff_ave,0.13);
172 + end
173 +
174 + if label==0 %bad
175 + disp(file_name(1:20));
176 + disp("difference");
177 + disp(target(tar_index));
178 + disp(diff_ave);
179 + end
180 + end
181 +
182 +
183 + % reds(head-spine) shoud be parallel to z
184 + % 4 3 21 2 1
185 +
186 + max_t=[];
187 +
188 + for j = 1: length(red)-1
189 + theta_per_patch=[];
190 + for patch = 1:frame_num
191 + u = [clear_subject_x(patch, red(j+1))-clear_subject_x(patch, red(j)),clear_subject_y(patch, red(j+1))-clear_subject_y(patch, red(j)),clear_subject_z(patch, red(j+1))-clear_subject_z(patch, red(j))];
192 + % v = axis z
193 + v = [0,0,1];
194 + % angle between [vector joint to joint] and [axis z]
195 + theta = atan2(norm(cross(u,v)),dot(u,v));
196 + %if theta > 90
197 + %theta = theta - 90;
198 + %end
199 +
200 + %print all the theta and max of it.
201 + %disp(theta);
202 +
203 + theta_per_patch=[theta_per_patch, theta];
204 + end
205 + % maximum theta for each bones through all the patches
206 + % max_t[4]
207 + max_t = [max_t,max(theta_per_patch)];
208 + clear theta_per_patch;
209 +
210 + end
211 + %disp("max");
212 + %disp(max_t);
213 + % usually 1.6<max angle<=1.8
214 + if label~=0
215 + label=set_label(max_t,1.8);
216 + end
217 + % green, blue(arms) should be located in almost the same position at the
218 + % beginning and end of motion.
219 +
220 +
221 + %print a distance between starting and ending coords
222 + g_distance=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,green);
223 + b_distance=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,blue);
224 +
225 +
226 + % only use just 2 joints at the tip of arms(hands)
227 + g_end=g_distance(end-1:end);
228 + b_end=b_distance(end-1:end);
229 +
230 + if label~=0
231 + label=set_label(g_end,0.7);
232 + end
233 +
234 + % check if label is still 'good'
235 + if label~=0
236 + label=set_label(b_end,0.7);
237 + end
238 +
239 +
240 + % save good and bad examples seperately
241 + newdir='';
242 + if label==0 % bad
243 + newdir='/home/hyuna/Documents/actionGAN_work/20_bad.txt';
244 + count(2)=count(2)+1;
245 + else
246 + newdir='/home/hyuna/Documents/actionGAN_work/20_good.txt';
247 + count(3)=count(3)+1;
248 + copyfile(name,'/home/hyuna/Documents/actionGAN_work/20_good_skeleton');
249 +
250 +
251 +
252 +
253 +
254 + split=fopen(s_dir,'a');
255 + fprintf(split, file_name(1:20));
256 + fprintf(split, '\n');
257 + fclose(split);
258 +
259 + end
260 +
261 + txtfile=fopen(newdir,'a');
262 + fprintf(txtfile, file_name(1:20));
263 + fprintf(txtfile, '\n');
264 + fclose(txtfile);
265 +
266 +
267 +end
268 +
269 +% number of [emptybody, bad, good]
270 +disp(count);
271 +
272 +function setlabel=set_label(target,value)
273 +if target(target>value) %bad
274 + label=0;
275 + else
276 + label=1;
277 +end
278 +setlabel=label;
279 +end
280 +
281 +
282 +function loc_end=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target)
283 +distance = []
284 + for j = 1: length(target)
285 + x_=clear_subject_x(1, target(j))-clear_subject_x(frame_num, target(j));
286 + y_=clear_subject_y(1, target(j))-clear_subject_y(frame_num, target(j));
287 + z_=clear_subject_z(1, target(j))-clear_subject_z(frame_num, target(j));
288 + d=sqrt(x_*x_+y_*y_+z_*z_);
289 + distance=[distance,d];
290 + end
291 +loc_end =distance;
292 +% disp(distance);
293 +end
294 +
295 +function comp_ave =comp_aver(clear_subject_x,clear_subject_y,clear_subject_z, frame_num,target)
296 +for j = 1: length(target)
297 + % distance between a particular joint and average coordinate per patch
298 + dist_ave=[];
299 + for patch = 2:frame_num-1
300 + ave_x=(clear_subject_x(patch-1,target(j))+clear_subject_x(patch+1,target(j)))/2;
301 + ave_y=(clear_subject_y(patch-1,target(j))+clear_subject_y(patch+1,target(j)))/2;
302 + ave_z=(clear_subject_z(patch-1,target(j))+clear_subject_z(patch+1,target(j)))/2;
303 +
304 + % distance between joint and average
305 + jnt_ave = sqrt((abs(clear_subject_x(patch, target(j))-ave_x)).^2+ (abs(clear_subject_y(patch, target(j))-ave_y)).^2+ (abs(clear_subject_z(patch, target(j))-ave_z)).^2);
306 + dist_ave=[dist_ave,jnt_ave];
307 +
308 + end
309 +
310 +end
311 + comp_ave=dist_ave;
312 +end
313 +function major_arm= arms(clear_subject_z, target1, target2)
314 + z1=max(clear_subject_z(:,target1(end))) - clear_subject_z(1, target1(end));
315 + z2=max(clear_subject_z(:,target2(end))) - clear_subject_z(1, target2(end));
316 +
317 + if z1>z2 % target1 raised higher than t2
318 + major_arm=target1;
319 + else
320 + major_arm=target2;
321 + end
322 +
323 +end
324 +
1 +
2 +% 맨처음 방향 (frame1) 으로 일괄 orientation normalize
3 +% => 1번프레임이 아니라 프레임들의 중앙값
4 +% 키로 모든방향 normalize
5 +% 노멀라이즈시 최소값 빼는게 아니라 joint1-> 0.5/0.5/0.5로 가도록
6 +% 실험할때 전방향 노멀라이즈도 해봐야겠다..
7 +path_name ='/home/hyuna/Documents/actionGAN_work/24_cleansed_skeleton/';
8 +dinfo=dir('/home/hyuna/Documents/actionGAN_work/24_cleansed_skeleton/*.skeleton');
9 +%txt=fullfile(dir_to_search, '*.skeleton');
10 +%dinfo=dir(txt);
11 +count=[0,0,0];
12 +for d = 1: length(dinfo)
13 + label=-1;
14 + file_name = dinfo(d).name;
15 + % disp(name);
16 +
17 + name = strcat(path_name,file_name(1:20),'.skeleton');
18 + [token,remainder] = strtok(file_name,'A');
19 +
20 + class = str2num(remainder(2:4));
21 + %class=remainder(2:4);
22 +
23 + if class == 24
24 + bodyinfo = read_skeleton_file(name);
25 + frame_num = size(bodyinfo,2);
26 +
27 + else
28 + continue;
29 + end
30 +
31 + if isempty(bodyinfo) || isempty(bodyinfo(1).bodies())
32 + disp("empty body");
33 + newdir='/home/hyuna/Documents/actionGAN_work/24_emptybody.txt';
34 +
35 + txtfile=fopen(newdir,'a');
36 + fprintf(txtfile, file_name(1:20));
37 + fprintf(txtfile, '\n');
38 + fclose(txtfile);
39 + count(1)=count(1)+1;
40 + continue;
41 + end
42 +
43 +
44 + %initialize
45 + cur_subject_x = zeros(frame_num, 25);
46 + cur_subject_y = zeros(frame_num, 25);
47 + cur_subject_z = zeros(frame_num, 25);
48 +
49 + tot_x = zeros(frame_num,25);
50 + tot_y = zeros(frame_num,25);
51 + tot_z = zeros(frame_num,25);
52 +
53 + joint_5 = zeros(1,3);
54 + joint_9 = zeros(1,3);
55 + joint_1 = zeros(1,3);
56 + joint_3 = zeros(1,3);
57 +
58 +
59 +
60 + %get total joints information
61 + for FN = 1:frame_num
62 +
63 + cur_body = bodyinfo(FN).bodies(1);
64 + joints = cur_body.joints;
65 +
66 + for JN = 1:25
67 + tot_x(FN,JN) = joints(JN).x;
68 + tot_y(FN,JN) = joints(JN).y;
69 + tot_z(FN,JN) = joints(JN).z;
70 + end
71 + end
72 +
73 + %Orientation normalization 1 : in space
74 + %get median values
75 + M_x = median(tot_x);
76 + M_y = median(tot_y);
77 + M_z = median(tot_z);
78 +
79 + %set 3 points for make plane
80 + joint_5 = [M_x(5) M_y(5) M_z(5)];
81 + joint_9 = [M_x(9) M_y(9) M_z(9)];
82 + joint_1 = [M_x(1) M_y(1) M_z(1)];
83 + joint_3 = [M_x(3) M_y(3) M_z(3)];
84 +
85 + %find RIGID TRNASFORMATION matrix
86 + d1 = joint_1 - joint_5;
87 + d2 = joint_1 - joint_9;
88 + n1 = cross(d1,d2); % because we will parallel transform, don't need to find belly
89 + u1 = n1/norm(n1);
90 + u2 = [0 0 1];
91 + cs1 = dot(u1,u2)/norm(u1)*norm(u2);
92 + ss1 = sqrt(1-cs1.^2);
93 + v1 = cross(u1,u2)/norm(cross(u1,u2));
94 +
95 + R1 = [v1(1)*v1(1)*(1-cs1)+cs1 v1(1)*v1(2)*(1-cs1)-v1(3)*ss1 v1(1)*v1(3)*(1-cs1)+v1(2)*ss1];
96 + R1(2,:) = [v1(1)*v1(2)*(1-cs1)+v1(3)*ss1 v1(2)*v1(2)*(1-cs1)+cs1 v1(2)*v1(3)*(1-cs1)-v1(1)*ss1];
97 + R1(3,:) = [v1(1)*v1(3)*(1-cs1)-v1(2)*ss1 v1(2)*v1(3)*(1-cs1)+v1(1)*ss1 v1(3)*v1(3)*(1-cs1)+cs1];
98 +
99 + %1-3 number tolls to parallel x axis. Rigid transformation on plane surface
100 + %Z axis coords oyler angle transform
101 +
102 + t = joint_3 - joint_1;
103 + d3 = R1(1,:) * t.';
104 + d3(1,2) = R1(2,:) * t.';
105 + d3(1,3) = R1(3,:) * t.';
106 +
107 + u3 = d3(1:2)/norm(d3(1:2));
108 + v3 = [u3(1) -u3(2)];
109 + v3(2,:) = [u3(2) u3(1)];
110 + u4 = [1 0].';
111 +
112 + csss = v3\u4;
113 + cs2 = csss(1);
114 + ss2 = csss(2);
115 +
116 + R2 = [cs2 -ss2 0];
117 + R2(2,:) = [ss2 cs2 0];
118 + R2(3,:) = [0 0 1];
119 +
120 +
121 + %apply rigid transformation
122 + for FN = 1:frame_num
123 + cur_body = bodyinfo(FN).bodies(1);
124 + joints = cur_body.joints;
125 +
126 + for JN = 1:25
127 + a = R1(1,:) * [joints(JN).x joints(JN).y joints(JN).z].';
128 + b = R1(2,:) * [joints(JN).x joints(JN).y joints(JN).z].';
129 + c = R1(3,:) * [joints(JN).x joints(JN).y joints(JN).z].';
130 +
131 + cur_subject_x(FN,JN) = R2(1,:) * [a b c].';
132 + cur_subject_y(FN,JN) = R2(2,:) * [a b c].';
133 + cur_subject_z(FN,JN) = R2(3,:) * [a b c].';
134 +
135 + end
136 + end
137 +
138 + %orientation normalize 2 (with plane surface)
139 + if cur_subject_x(1,4) < cur_subject_x(1,1)
140 + cur_subject_x = 0 - cur_subject_x;
141 + end
142 +
143 + if cur_subject_y(1,9) > cur_subject_y(1,5)
144 + cur_subject_y = 0 - cur_subject_y;
145 + end
146 +
147 + % for save origin subjects before data augment
148 + clear_subject_x = cur_subject_x;
149 + clear_subject_y = cur_subject_y;
150 + clear_subject_z = cur_subject_z;
151 +
152 + % patch_num*25
153 + red=[4,3,21,2,1];
154 + green=[5,6,7,8,23,22];
155 + blue=[9,10,11,12,25,24];
156 + skyblue=[17,18,19,20];
157 + yellow=[13,14,15,16];
158 +
159 + target = {skyblue,yellow,green,blue};
160 + [r,c]=size(target);
161 +
162 + %celldisp(target);
163 + %disp(c);
164 + %disp(target{1});
165 +
166 + % joint-to-joint
167 + for tar_index = 1:c
168 + diff_ave = comp_aver(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target{tar_index});
169 +
170 + if label~=0
171 + label=set_label(diff_ave,0.13);
172 + end
173 +
174 + if label==0 %bad
175 + disp(file_name(1:20));
176 + disp("difference");
177 + disp(target(tar_index));
178 + disp(diff_ave);
179 + end
180 + end
181 +
182 +
183 + % reds(head-spine) shoud be parallel to z
184 + % 4 3 21 2 1
185 +
186 + max_t=[];
187 +
188 + for j = 1: length(red)-1
189 + theta_per_patch=[];
190 + for patch = 1:frame_num
191 + u = [clear_subject_x(patch, red(j+1))-clear_subject_x(patch, red(j)),clear_subject_y(patch, red(j+1))-clear_subject_y(patch, red(j)),clear_subject_z(patch, red(j+1))-clear_subject_z(patch, red(j))];
192 + % v = axis z
193 + v = [0,0,1];
194 + % angle between [vector joint to joint] and [axis z]
195 + theta = atan2(norm(cross(u,v)),dot(u,v));
196 + %if theta > 90
197 + %theta = theta - 90;
198 + %end
199 +
200 + %print all the theta and max of it.
201 + %disp(theta);
202 +
203 + theta_per_patch=[theta_per_patch, theta];
204 + end
205 + % maximum theta for each bones through all the patches
206 + % max_t[4]
207 + max_t = [max_t,max(theta_per_patch)];
208 + clear theta_per_patch;
209 +
210 + end
211 + %disp("max");
212 + %disp(max_t);
213 + % usually 1.6<max angle<=1.8
214 + if label~=0
215 + label=set_label(max_t,1.8);
216 + end
217 + % yellow, skyblue(legs) should be located in almost the same position at the
218 + % beginning and end of motion.
219 +
220 +
221 + %print a distance between starting and ending coords
222 + g_distance=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,yellow);
223 + b_distance=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,skyblue);
224 +
225 +
226 + % only use just 2 joints at the tip of legs(feet)
227 + g_end=g_distance(end-1:end);
228 + b_end=b_distance(end-1:end);
229 +
230 + if label~=0
231 + label=set_label(g_end,0.7);
232 + end
233 +
234 + % check if label is still 'good'
235 + if label~=0
236 + label=set_label(b_end,0.7);
237 + end
238 +
239 +
240 + % save good and bad examples seperately
241 + newdir='';
242 + if label==0 % bad
243 + newdir='/home/hyuna/Documents/actionGAN_work/24_bad.txt';
244 + count(2)=count(2)+1;
245 + else
246 + newdir='/home/hyuna/Documents/actionGAN_work/24_good.txt';
247 + count(3)=count(3)+1;
248 + copyfile(name,'/home/hyuna/Documents/actionGAN_work/24_good_skeleton');
249 +
250 + major =arms(clear_subject_z, yellow, skyblue);
251 + if(major==yellow)
252 + s_dir='/home/hyuna/Documents/actionGAN_work/good_right.txt';
253 + copyfile(name,'/home/hyuna/Documents/actionGAN_work/right_skeleton');
254 + else
255 + s_dir='/home/hyuna/Documents/actionGAN_work/good_left.txt';
256 + copyfile(name,'/home/hyuna/Documents/actionGAN_work/left_skeleton');
257 + end
258 +
259 +
260 + split=fopen(s_dir,'a');
261 + fprintf(split, file_name(1:20));
262 + fprintf(split, '\n');
263 + fclose(split);
264 +
265 + end
266 +
267 + txtfile=fopen(newdir,'a');
268 + fprintf(txtfile, file_name(1:20));
269 + fprintf(txtfile, '\n');
270 + fclose(txtfile);
271 +
272 +
273 +end
274 +
275 +% number of [emptybody, bad, good]
276 +disp(count);
277 +
278 +function setlabel=set_label(target,value)
279 +if target(target>value) %bad
280 + label=0;
281 + else
282 + label=1;
283 +end
284 +setlabel=label;
285 +end
286 +
287 +
288 +function loc_end=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target)
289 +distance = []
290 + for j = 1: length(target)
291 + x_=clear_subject_x(1, target(j))-clear_subject_x(frame_num, target(j));
292 + y_=clear_subject_y(1, target(j))-clear_subject_y(frame_num, target(j));
293 + z_=clear_subject_z(1, target(j))-clear_subject_z(frame_num, target(j));
294 + d=sqrt(x_*x_+y_*y_+z_*z_);
295 + distance=[distance,d];
296 + end
297 +loc_end =distance;
298 +% disp(distance);
299 +end
300 +
301 +function comp_ave =comp_aver(clear_subject_x,clear_subject_y,clear_subject_z, frame_num,target)
302 +for j = 1: length(target)
303 + % distance between a particular joint and average coordinate per patch
304 + dist_ave=[];
305 + for patch = 2:frame_num-1
306 + ave_x=(clear_subject_x(patch-1,target(j))+clear_subject_x(patch+1,target(j)))/2;
307 + ave_y=(clear_subject_y(patch-1,target(j))+clear_subject_y(patch+1,target(j)))/2;
308 + ave_z=(clear_subject_z(patch-1,target(j))+clear_subject_z(patch+1,target(j)))/2;
309 +
310 + % distance between joint and average
311 + jnt_ave = sqrt((abs(clear_subject_x(patch, target(j))-ave_x)).^2+ (abs(clear_subject_y(patch, target(j))-ave_y)).^2+ (abs(clear_subject_z(patch, target(j))-ave_z)).^2);
312 + dist_ave=[dist_ave,jnt_ave];
313 +
314 + end
315 +
316 +end
317 + comp_ave=dist_ave;
318 +end
319 +function major_arm= arms(clear_subject_z, target1, target2)
320 + z1=max(clear_subject_z(:,target1(end))) - clear_subject_z(1, target1(end));
321 + z2=max(clear_subject_z(:,target2(end))) - clear_subject_z(1, target2(end));
322 +
323 + if z1>z2 % target1 raised higher than t2
324 + major_arm=target1;
325 + else
326 + major_arm=target2;
327 + end
328 +
329 +end
330 +
This diff is collapsed. Click to expand it.