def save_as(self, event): if self.f_name_tb.GetValue() != '': sets = self.img_panel.get_ranges() if (len(sets[1]) + len(sets[2]) + len(sets[3]) != 6): erwr = wx.MessageDialog(self,'Ranges are not properly selected.' 'Ranges Malformed', wx.OK) erwr.ShowModal() erwr.Destroy() return #write an entry to the replay log replay_file = open('./replay.txt','a') replay_file.write(self.img_file + '\n') replay_file.write('1,' + str(sets[1][0][0])+','+str(sets[1][0][1])+','+ str(sets[1][1][0]-sets[1][0][0])+','+str(sets[1][1][1]-sets[1][0][1])+'\n') replay_file.write('2,' + str(sets[2][0][0])+','+str(sets[2][0][1])+','+ str(sets[2][1][0]-sets[2][0][0])+','+str(sets[2][1][1]-sets[2][0][1])+'\n') replay_file.write('3,' + str(sets[3][0][0])+','+str(sets[3][0][1])+','+ str(sets[3][1][0]-sets[3][0][0])+','+str(sets[3][1][1]-sets[3][0][1])+'\n') replay_file.write(str(self.typebox.GetSelection()) + '\n') replay_file.write(self.f_name_tb.GetValue() + '\n') replay_file.close() f_s = face_sampler() left_eye_range = (sets[1][0][0],sets[1][0][1],sets[1][1][0]-sets[1][0][0],sets[1][1][1]-sets[1][0][1]) right_eye_range = (sets[2][0][0],sets[2][0][1],sets[2][1][0]-sets[2][0][0],sets[2][1][1]-sets[2][0][1]) mouth_range = (sets[3][0][0],sets[3][0][1],sets[3][1][0]-sets[3][0][0],sets[3][1][1]-sets[3][0][1]) p_s = f_s.gen_pixel_set(self.img,left_eye_range,right_eye_range,mouth_range) f_s.write_to_csv(p_s,self.typebox.GetSelection(),self.f_name_tb.GetValue())
def __init__(self,replay_file): self.f_s = face_sampler() self.replays = [] rf = open(replay_file,'r') while True: rp_file = rf.readline().strip() if rp_file == None or rp_file == '': break l_e = rf.readline().strip().split(',') r_e = rf.readline().strip().split(',') mouth = rf.readline().strip().split(',') type = rf.readline().strip() ofile = rf.readline().strip() self.replays.append((rp_file,(int(l_e[1]),int(l_e[2]),int(l_e[3]),int(l_e[4])), (int(r_e[1]),int(r_e[2]),int(r_e[3]),int(r_e[4])), (int(mouth[1]),int(mouth[2]),int(mouth[3]),int(mouth[4])), type,ofile))
def save_as(self, event): if self.f_name_tb.GetValue() != '': sets = self.img_panel.get_ranges() if (len(sets[1]) + len(sets[2]) + len(sets[3]) != 6): erwr = wx.MessageDialog( self, 'Ranges are not properly selected.' 'Ranges Malformed', wx.OK) erwr.ShowModal() erwr.Destroy() return #write an entry to the replay log replay_file = open('./replay.txt', 'a') replay_file.write(self.img_file + '\n') replay_file.write('1,' + str(sets[1][0][0]) + ',' + str(sets[1][0][1]) + ',' + str(sets[1][1][0] - sets[1][0][0]) + ',' + str(sets[1][1][1] - sets[1][0][1]) + '\n') replay_file.write('2,' + str(sets[2][0][0]) + ',' + str(sets[2][0][1]) + ',' + str(sets[2][1][0] - sets[2][0][0]) + ',' + str(sets[2][1][1] - sets[2][0][1]) + '\n') replay_file.write('3,' + str(sets[3][0][0]) + ',' + str(sets[3][0][1]) + ',' + str(sets[3][1][0] - sets[3][0][0]) + ',' + str(sets[3][1][1] - sets[3][0][1]) + '\n') replay_file.write(str(self.typebox.GetSelection()) + '\n') replay_file.write(self.f_name_tb.GetValue() + '\n') replay_file.close() f_s = face_sampler() left_eye_range = (sets[1][0][0], sets[1][0][1], sets[1][1][0] - sets[1][0][0], sets[1][1][1] - sets[1][0][1]) right_eye_range = (sets[2][0][0], sets[2][0][1], sets[2][1][0] - sets[2][0][0], sets[2][1][1] - sets[2][0][1]) mouth_range = (sets[3][0][0], sets[3][0][1], sets[3][1][0] - sets[3][0][0], sets[3][1][1] - sets[3][0][1]) p_s = f_s.gen_pixel_set(self.img, left_eye_range, right_eye_range, mouth_range) f_s.write_to_csv(p_s, self.typebox.GetSelection(), self.f_name_tb.GetValue())