조현아

filter by motionpatch

path_name ='C:\\Users\\User\\Documents\\gan_work\\25\\';
dinfo=dir('C:\\Users\\User\\Documents\\gan_work\\25\\*.png');
%txt=fullfile(dir_to_search, '*.png');
%dinfo=dir(txt);
%disp(cur_subject_x);
count=[0,0,0];
for d = 1: length(dinfo)
label=-1;
file_name = dinfo(d).name;
motionpatch=imread(fullfile(path_name, file_name));
cur_subject_x=double(motionpatch(:,:,1))./255;
cur_subject_y=double(motionpatch(:,:,2))./255;
cur_subject_z=double(motionpatch(:,:,3))./255;
frame_num = size(cur_subject_x,1);
name = strcat(path_name,file_name(1:length(file_name)),'.png');
% for save origin subjects before data augment
clear_subject_x = cur_subject_x;
clear_subject_y = cur_subject_y;
clear_subject_z = cur_subject_z;
% patch_num*25
red=[4,3,21,2,1];
green=[5,6,7,8,23,22];
blue=[9,10,11,12,25,24];
yellow=[17,18,19,20];
skyblue=[13,14,15,16];
target = {skyblue,yellow,green,blue};
[r,c]=size(target);
%celldisp(target);
%disp(c);
%disp(target{1});
% joint-to-joint
for tar_index = 1:c
diff_ave = comp_aver(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target{tar_index});
if label~=0
label=set_label(diff_ave,0.13);
end
%{
if label==0 %bad
disp(file_name(1:20));
disp("difference");
disp(tar_index);
disp(target(tar_index));
disp(diff_ave);
end
%}
end
% reds(head-spine) shoud be always parallel to z
if label~=0
label=set_label(stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, 1,frame_num,red),1.8);
end
% should check straightness for both legs in case 27 only for start and
% end of execution
%{
if label~=0
disp(name);
disp(stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, 1,3, skyblue));
disp(stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, frame_num-3,frame_num, skyblue));
end
%}
for tar_index = 1:c
label= set_label(stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, 1,3, target{tar_index}),2.0);
if label==0
break;
end
label= set_label(stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, frame_num-3,frame_num, target{tar_index}),2.0);
if label==0
break;
end
end
% save good and bad examples seperately
newdir='';
if label==0 % bad
newdir='C:\\Users\\User\\Documents\\gan_work\\datas\\mp_bad_1.1.txt';
count(2)=count(2)+1;
%copyfile(name,'C:\\Users\\User\\Documents\\gan_work\\datas\\mp_bad');
else % good
newdir='C:\\Users\\User\\Documents\\gan_work\\datas\\mp_good_1.1.txt';
count(3)=count(3)+1;
%copyfile(name,'C:\\Users\\User\\Documents\\gan_work\\datas\\mp_good');
% The condition below is used for a class with subclasses of right/left major limb.
%{
major =major_limb(clear_subject_z, skyblue, yellow);
if(major==skyblue)
s_dir='/home/hyuna/Documents/actionGAN_work/24/24_good_right.txt';
copyfile(name,'/home/hyuna/Documents/actionGAN_work/24/24_right_skeleton');
else
s_dir='/home/hyuna/Documents/actionGAN_work/24/24_good_left.txt';
copyfile(name,'/home/hyuna/Documents/actionGAN_work/24/24_left_skeleton');
end
%}
% The condition below is used for a class 27 to classify how many
% times it jumped.
end
txtfile=fopen(newdir,'a');
fprintf(txtfile,file_name(1:length(file_name)));
fprintf(txtfile, '\n');
fclose(txtfile);
end
% number of [emptybody, bad, good]
disp(count);
function stand=stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, start,end_ind,target)
max_t=[];
for j = 1: length(target)-1
theta_per_patch=[];
for patch = start:end_ind % usually 1:frame_num
u = [clear_subject_x(patch, target(j+1))-clear_subject_x(patch, target(j)),clear_subject_y(patch, target(j+1))-clear_subject_y(patch, target(j)),clear_subject_z(patch, target(j+1))-clear_subject_z(patch, target(j))];
% v = axis z
v = [0,0,1];
% angle between [vector joint to joint] and [axis z]
theta = atan2(norm(cross(u,v)),dot(u,v));
%if theta > 90
%theta = theta - 90;
%end
%print all the theta and max of it.
%disp(theta);
theta_per_patch=[theta_per_patch, theta];
end
% maximum theta for each bones through all the patches
% max_t[4]
max_t = [max_t,max(theta_per_patch)];
clear theta_per_patch;
end
%disp("max");
%disp(max_t);
% usually 1.6<max angle<=1.8
stand=max_t;
end
function setlabel=set_label(target,value)
value=value*1.1;
if target(target>value) %bad
label=0;
else
label=1;
end
setlabel=label;
end
function loc_end=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target)
echo off;
distance=[];
for j = 1: length(target)
x_=clear_subject_x(1, target(j))-clear_subject_x(frame_num, target(j));
y_=clear_subject_y(1, target(j))-clear_subject_y(frame_num, target(j));
z_=clear_subject_z(1, target(j))-clear_subject_z(frame_num, target(j));
d=sqrt(x_*x_+y_*y_+z_*z_);
distance=[distance,d];
end
loc_end =distance;
% disp(distance);
end
function comp_ave =comp_aver(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target)
for j = 1: length(target)
% distance between a particular joint and average coordinate per patch
dist_ave=[];
for patch = 2:frame_num-1
% disp(patch);
ave_x=(clear_subject_x(patch-1,target(j))+clear_subject_x(patch+1,target(j)))/2;
ave_y=(clear_subject_y(patch-1,target(j))+clear_subject_y(patch+1,target(j)))/2;
ave_z=(clear_subject_z(patch-1,target(j))+clear_subject_z(patch+1,target(j)))/2;
% distance between joint and average
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);
dist_ave=[dist_ave,jnt_ave];
end
end
comp_ave=dist_ave;
end
function limb= major_limb(clear_subject_z, target1, target2)
z1=max(clear_subject_z(:,target1(end))) - clear_subject_z(1, target1(end));
z2=max(clear_subject_z(:,target2(end))) - clear_subject_z(1, target2(end));
if z1>z2 % target1 raised higher than t2
limb=target1;
else
limb=target2;
end
end
function num = num_jumped(clear_subject_z, target,frame_num)
state=0; %
for i = 1: frame_num-1
% check if it had descended
if state==1 && clear_subject_z(i,target) < clear_subject_z(i+1,target)
state=2;
elseif clear_subject_z(i,target)>clear_subject_z(i+1,target)
state=1; % first descending
else
state=-1;
end
end
num=state;
end
path_name ='C:\\Users\\User\\Documents\\gan_work\\25\\';
dinfo=dir('C:\\Users\\User\\Documents\\gan_work\\25\\*.png');
count=[0,0,0];
for d = 1: length(dinfo)
label=-1;
file_name = dinfo(d).name;
motionpatch=imread(fullfile(path_name, file_name));
cur_subject_x=double(motionpatch(:,:,1))./255;
cur_subject_y=double(motionpatch(:,:,2))./255;
cur_subject_z=double(motionpatch(:,:,3))./255;
frame_num = size(cur_subject_x,1);
name = strcat(path_name,file_name(1:length(file_name)),'.png');
% for save origin subjects before data augment
clear_subject_x = cur_subject_x;
clear_subject_y = cur_subject_y;
clear_subject_z = cur_subject_z;
% patch_num*25
red=[4,3,21,2,1];
green=[5,6,7,8,23,22];
blue=[9,10,11,12,25,24];
skyblue=[17,18,19,20];
yellow=[13,14,15,16];
target = {skyblue,yellow,green,blue};
[r,c]=size(target);
%celldisp(target);
%disp(c);
%disp(target{1});
% joint-to-joint
for tar_index = 1:c
diff_ave = comp_aver(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target{tar_index});
if label~=0
label=set_label(diff_ave,0.13);
end
if label==0 %bad
disp(file_name(1:20));
disp("difference");
disp(target(tar_index));
disp(diff_ave);
end
end
% reds(head-spine) shoud be parallel to z
% 4 3 21 2 1
max_t=[];
for j = 1: length(red)-1
theta_per_patch=[];
for patch = 1:frame_num
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))];
% v = axis z
v = [0,0,1];
% angle between [vector joint to joint] and [axis z]
theta = atan2(norm(cross(u,v)),dot(u,v));
%if theta > 90
%theta = theta - 90;
%end
%print all the theta and max of it.
%disp(theta);
theta_per_patch=[theta_per_patch, theta];
end
% maximum theta for each bones through all the patches
% max_t[4]
max_t = [max_t,max(theta_per_patch)];
clear theta_per_patch;
end
%disp("max");
%disp(max_t);
% usually 1.6<max angle<=1.8
if label~=0
label=set_label(max_t,1.8);
end
% green, blue(arms) should be located in almost the same position at the
% beginning and end of motion.
%print a distance between starting and ending coords
g_distance=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,yellow);
b_distance=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,skyblue);
% only use just 2 joints at the tip of arms(hands)
g_end=g_distance(end-1:end);
b_end=b_distance(end-1:end);
if label~=0
label=set_label(g_end,0.7);
end
% check if label is still 'good'
if label~=0
label=set_label(b_end,0.7);
end
% save good and bad examples seperately
newdir='';
if label==0 % bad
newdir='/home/hyuna/Documents/actionGAN_work/7_bad.txt';
count(2)=count(2)+1;
else
newdir='/home/hyuna/Documents/actionGAN_work/7_good.txt';
count(3)=count(3)+1;
%copyfile(name,'/home/hyuna/Documents/actionGAN_work/7_good_skeleton');
major =arms(clear_subject_z, yellow, skyblue);
if(major==yellow)
s_dir='/home/hyuna/Documents/actionGAN_work/good_right.txt';
copyfile(name,'/home/hyuna/Documents/actionGAN_work/right_skeleton');
else
s_dir='/home/hyuna/Documents/actionGAN_work/good_left.txt';
copyfile(name,'/home/hyuna/Documents/actionGAN_work/left_skeleton');
end
split=fopen(s_dir,'a');
fprintf(split, file_name(1:20));
fprintf(split, '\n');
fclose(split);
end
txtfile=fopen(newdir,'a');
fprintf(txtfile, file_name(1:20));
fprintf(txtfile, '\n');
fclose(txtfile);
end
% number of [emptybody, bad, good]
disp(count);
function setlabel=set_label(target,value)
value=value*1.1;
if target(target>value) %bad
label=0;
else
label=1;
end
setlabel=label;
end
function loc_end=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target)
distance = []
for j = 1: length(target)
x_=clear_subject_x(1, target(j))-clear_subject_x(frame_num, target(j));
y_=clear_subject_y(1, target(j))-clear_subject_y(frame_num, target(j));
z_=clear_subject_z(1, target(j))-clear_subject_z(frame_num, target(j));
d=sqrt(x_*x_+y_*y_+z_*z_);
distance=[distance,d];
end
loc_end =distance;
% disp(distance);
end
function comp_ave =comp_aver(clear_subject_x,clear_subject_y,clear_subject_z, frame_num,target)
for j = 1: length(target)
% distance between a particular joint and average coordinate per patch
dist_ave=[];
for patch = 2:frame_num-1
ave_x=(clear_subject_x(patch-1,target(j))+clear_subject_x(patch+1,target(j)))/2;
ave_y=(clear_subject_y(patch-1,target(j))+clear_subject_y(patch+1,target(j)))/2;
ave_z=(clear_subject_z(patch-1,target(j))+clear_subject_z(patch+1,target(j)))/2;
% distance between joint and average
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);
dist_ave=[dist_ave,jnt_ave];
end
end
comp_ave=dist_ave;
end
function major_arm= arms(clear_subject_z, target1, target2)
z1=max(clear_subject_z(:,target1(end))) - clear_subject_z(1, target1(end));
z2=max(clear_subject_z(:,target2(end))) - clear_subject_z(1, target2(end));
if z1>z2 % target1 raised higher than t2
major_arm=target1;
else
major_arm=target2;
end
end
path_name ='C:\\Users\\User\\Documents\\gan_work\\25\\';
dinfo=dir('C:\\Users\\User\\Documents\\gan_work\\25\\*.png');
%txt=fullfile(dir_to_search, '*.png');
%dinfo=dir(txt);
%disp(cur_subject_x);
count=[0,0,0];
for d = 1: length(dinfo)
label=-1;
file_name = dinfo(d).name;
motionpatch=imread(fullfile(path_name, file_name));
cur_subject_x=double(motionpatch(:,:,1))./255;
cur_subject_y=double(motionpatch(:,:,2))./255;
cur_subject_z=double(motionpatch(:,:,3))./255;
frame_num = size(cur_subject_x,1);
name = strcat(path_name,file_name(1:length(file_name)),'.png');
% for save origin subjects before data augment
clear_subject_x = cur_subject_x;
clear_subject_y = cur_subject_y;
clear_subject_z = cur_subject_z;
% patch_num*25
red=[4,3,21,2,1];
green=[5,6,7,8,23,22];
blue=[9,10,11,12,25,24];
yellow=[17,18,19,20];
skyblue=[13,14,15,16];
target = {skyblue,yellow,green,blue};
[r,c]=size(target);
%celldisp(target);
%disp(c);
%disp(target{1});
% joint-to-joint
for tar_index = 1:c
diff_ave = comp_aver(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target{tar_index});
if label~=0
label=set_label(diff_ave,0.13);
end
%{
if label==0 %bad
disp(file_name(1:20));
disp("difference");
disp(tar_index);
disp(target(tar_index));
disp(diff_ave);
end
%}
end
% reds(head-spine) shoud be always parallel to z
if label~=0
label=set_label(stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, 1,frame_num,red),1.8);
end
% should check straightness for both legs in case 27 only for start and
% end of execution
%{
if label~=0
disp(name);
disp(stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, 1,3, skyblue));
disp(stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, frame_num-3,frame_num, skyblue));
end
%}
for tar_index = 1:c
label= set_label(stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, 1,3, target{tar_index}),2.0);
if label==0
break;
end
label= set_label(stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, frame_num-3,frame_num, target{tar_index}),2.0);
if label==0
break;
end
end
% green, blue(arms) should be located in almost the same position at the
% beginning and end of motion.
%print a distance between starting and ending coords
g_distance=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,skyblue);
b_distance=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,yellow);
% only use just 2 joints at the tip of arms(hands)
g_end=g_distance(end-1:end);
b_end=b_distance(end-1:end);
if label~=0
label=set_label(g_end,0.7);
end
% check if label is still 'good'
if label~=0
label=set_label(b_end,0.7);
end
ju1=0;
ju2=0;
if label~=0
legs={skyblue, yellow};
% count how many times it jumped for each leg
ju1=num_jumped(clear_subject_z,legs{1}(2),frame_num);
ju2=num_jumped(clear_subject_z,legs{2}(2),frame_num);
if ju1 ~= ju2
label = 0;
end
end
% save good and bad examples seperately
newdir='';
if label==0 % bad
newdir='C:\\Users\\User\\Documents\\gan_work\\datas\\mp_bad.txt';
count(2)=count(2)+1;
%copyfile(name,'C:\\Users\\User\\Documents\\gan_work\\datas\\mp_bad');
else % good
newdir='C:\\Users\\User\\Documents\\gan_work\\datas\\mp_good.txt';
count(3)=count(3)+1;
%copyfile(name,'C:\\Users\\User\\Documents\\gan_work\\datas\\mp_good');
% The condition below is used for a class with subclasses of right/left major limb.
%{
major =major_limb(clear_subject_z, skyblue, yellow);
if(major==skyblue)
s_dir='/home/hyuna/Documents/actionGAN_work/24/24_good_right.txt';
copyfile(name,'/home/hyuna/Documents/actionGAN_work/24/24_right_skeleton');
else
s_dir='/home/hyuna/Documents/actionGAN_work/24/24_good_left.txt';
copyfile(name,'/home/hyuna/Documents/actionGAN_work/24/24_left_skeleton');
end
%}
% The condition below is used for a class 27 to classify how many
% times it jumped.
end
txtfile=fopen(newdir,'a');
fprintf(txtfile,file_name(1:length(file_name)));
fprintf(txtfile, '\n');
fclose(txtfile);
end
% number of [emptybody, bad, good]
disp(count);
function stand=stand_straight(clear_subject_x,clear_subject_y,clear_subject_z, start,end_ind,target)
max_t=[];
for j = 1: length(target)-1
theta_per_patch=[];
for patch = start:end_ind % usually 1:frame_num
u = [clear_subject_x(patch, target(j+1))-clear_subject_x(patch, target(j)),clear_subject_y(patch, target(j+1))-clear_subject_y(patch, target(j)),clear_subject_z(patch, target(j+1))-clear_subject_z(patch, target(j))];
% v = axis z
v = [0,0,1];
% angle between [vector joint to joint] and [axis z]
theta = atan2(norm(cross(u,v)),dot(u,v));
%if theta > 90
%theta = theta - 90;
%end
%print all the theta and max of it.
%disp(theta);
theta_per_patch=[theta_per_patch, theta];
end
% maximum theta for each bones through all the patches
% max_t[4]
max_t = [max_t,max(theta_per_patch)];
clear theta_per_patch;
end
%disp("max");
%disp(max_t);
% usually 1.6<max angle<=1.8
stand=max_t;
end
function setlabel=set_label(target,value)
value=value*1.1;
if target(target>value) %bad
label=0;
else
label=1;
end
setlabel=label;
end
function loc_end=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target)
echo off;
distance=[];
for j = 1: length(target)
x_=clear_subject_x(1, target(j))-clear_subject_x(frame_num, target(j));
y_=clear_subject_y(1, target(j))-clear_subject_y(frame_num, target(j));
z_=clear_subject_z(1, target(j))-clear_subject_z(frame_num, target(j));
d=sqrt(x_*x_+y_*y_+z_*z_);
distance=[distance,d];
end
loc_end =distance;
% disp(distance);
end
function comp_ave =comp_aver(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target)
for j = 1: length(target)
% distance between a particular joint and average coordinate per patch
dist_ave=[];
for patch = 2:frame_num-1
% disp(patch);
ave_x=(clear_subject_x(patch-1,target(j))+clear_subject_x(patch+1,target(j)))/2;
ave_y=(clear_subject_y(patch-1,target(j))+clear_subject_y(patch+1,target(j)))/2;
ave_z=(clear_subject_z(patch-1,target(j))+clear_subject_z(patch+1,target(j)))/2;
% distance between joint and average
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);
dist_ave=[dist_ave,jnt_ave];
end
end
comp_ave=dist_ave;
end
function limb= major_limb(clear_subject_z, target1, target2)
z1=max(clear_subject_z(:,target1(end))) - clear_subject_z(1, target1(end));
z2=max(clear_subject_z(:,target2(end))) - clear_subject_z(1, target2(end));
if z1>z2 % target1 raised higher than t2
limb=target1;
else
limb=target2;
end
end
function num = num_jumped(clear_subject_z, target,frame_num)
state=0; %
for i = 1: frame_num-1
% check if it had descended
if state==1 && clear_subject_z(i,target) < clear_subject_z(i+1,target)
state=2;
elseif clear_subject_z(i,target)>clear_subject_z(i+1,target)
state=1; % first descending
else
state=-1;
end
end
num=state;
end
path_name ='C:\\Users\\User\\Documents\\gan_work\\25\\';
dinfo=dir('C:\\Users\\User\\Documents\\gan_work\\25\\*.png');
count=[0,0,0];
for d = 1: length(dinfo)
label=-1;
file_name = dinfo(d).name;
motionpatch=imread(fullfile(path_name, file_name));
cur_subject_x=double(motionpatch(:,:,1))./255;
cur_subject_y=double(motionpatch(:,:,2))./255;
cur_subject_z=double(motionpatch(:,:,3))./255;
frame_num = size(cur_subject_x,1);
name = strcat(path_name,file_name(1:length(file_name)),'.png');
% for save origin subjects before data augment
clear_subject_x = cur_subject_x;
clear_subject_y = cur_subject_y;
clear_subject_z = cur_subject_z;
% patch_num*25
red=[4,3,21,2,1];
green=[5,6,7,8,23,22];
blue=[9,10,11,12,25,24];
skyblue=[17,18,19,20];
yellow=[13,14,15,16];
target = {skyblue,yellow,green,blue};
[r,c]=size(target);
%celldisp(target);
%disp(c);
%disp(target{1});
% joint-to-joint
for tar_index = 1:c
diff_ave = comp_aver(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target{tar_index});
if label~=0
label=set_label(diff_ave,0.13);
end
if label==0 %bad
disp(file_name(1:20));
disp("difference");
disp(target(tar_index));
disp(diff_ave);
end
end
% reds(head-spine) shoud be parallel to z
% 4 3 21 2 1
max_t=[];
for j = 1: length(red)-1
theta_per_patch=[];
for patch = 1:frame_num
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))];
% v = axis z
v = [0,0,1];
% angle between [vector joint to joint] and [axis z]
theta = atan2(norm(cross(u,v)),dot(u,v));
%if theta > 90
%theta = theta - 90;
%end
%print all the theta and max of it.
%disp(theta);
theta_per_patch=[theta_per_patch, theta];
end
% maximum theta for each bones through all the patches
% max_t[4]
max_t = [max_t,max(theta_per_patch)];
clear theta_per_patch;
end
%disp("max");
%disp(max_t);
% usually 1.6<max angle<=1.8
if label~=0
label=set_label(max_t,1.8);
end
% green, blue(arms) should be located in almost the same position at the
% beginning and end of motion.
%print a distance between starting and ending coords
g_distance=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,green);
b_distance=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,blue);
% only use just 2 joints at the tip of arms(hands)
g_end=g_distance(end-1:end);
b_end=b_distance(end-1:end);
if label~=0
label=set_label(g_end,0.7);
end
% check if label is still 'good'
if label~=0
label=set_label(b_end,0.7);
end
% save good and bad examples seperately
newdir='';
if label==0 % bad
newdir='/home/hyuna/Documents/actionGAN_work/7_bad.txt';
count(2)=count(2)+1;
else
newdir='/home/hyuna/Documents/actionGAN_work/7_good.txt';
count(3)=count(3)+1;
%copyfile(name,'/home/hyuna/Documents/actionGAN_work/7_good_skeleton');
major =arms(clear_subject_z, green, blue);
if(major==green)
s_dir='/home/hyuna/Documents/actionGAN_work/good_right.txt';
copyfile(name,'/home/hyuna/Documents/actionGAN_work/right_skeleton');
else
s_dir='/home/hyuna/Documents/actionGAN_work/good_left.txt';
copyfile(name,'/home/hyuna/Documents/actionGAN_work/left_skeleton');
end
split=fopen(s_dir,'a');
fprintf(split, file_name(1:20));
fprintf(split, '\n');
fclose(split);
end
txtfile=fopen(newdir,'a');
fprintf(txtfile, file_name(1:20));
fprintf(txtfile, '\n');
fclose(txtfile);
end
% number of [emptybody, bad, good]
disp(count);
function setlabel=set_label(target,value)
value=value*1.1;
if target(target>value) %bad
label=0;
else
label=1;
end
setlabel=label;
end
function loc_end=loc_end_to_end(clear_subject_x,clear_subject_y,clear_subject_z,frame_num,target)
distance = []
for j = 1: length(target)
x_=clear_subject_x(1, target(j))-clear_subject_x(frame_num, target(j));
y_=clear_subject_y(1, target(j))-clear_subject_y(frame_num, target(j));
z_=clear_subject_z(1, target(j))-clear_subject_z(frame_num, target(j));
d=sqrt(x_*x_+y_*y_+z_*z_);
distance=[distance,d];
end
loc_end =distance;
% disp(distance);
end
function comp_ave =comp_aver(clear_subject_x,clear_subject_y,clear_subject_z, frame_num,target)
for j = 1: length(target)
% distance between a particular joint and average coordinate per patch
dist_ave=[];
for patch = 2:frame_num-1
ave_x=(clear_subject_x(patch-1,target(j))+clear_subject_x(patch+1,target(j)))/2;
ave_y=(clear_subject_y(patch-1,target(j))+clear_subject_y(patch+1,target(j)))/2;
ave_z=(clear_subject_z(patch-1,target(j))+clear_subject_z(patch+1,target(j)))/2;
% distance between joint and average
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);
dist_ave=[dist_ave,jnt_ave];
end
end
comp_ave=dist_ave;
end
function major_arm= arms(clear_subject_z, target1, target2)
z1=max(clear_subject_z(:,target1(end))) - clear_subject_z(1, target1(end));
z2=max(clear_subject_z(:,target2(end))) - clear_subject_z(1, target2(end));
if z1>z2 % target1 raised higher than t2
major_arm=target1;
else
major_arm=target2;
end
end