MCPcopy Create free account
hub / github.com/OpenPTrack/open_ptrack_v2 / loadButtonCallback

Method loadButtonCallback

recognition/scripts/drag_and_drop.py:178–214  ·  view source on GitHub ↗
(self, e)

Source from the content-addressed store, hash-verified

176
177 # callback for load button pushing event
178 def loadButtonCallback(self, e):
179 self.setProgress(25, 'choose load filename')
180 load_filename = QtGui.QFileDialog.getOpenFileName()
181 if len(load_filename) == 0:
182 self.setProgress(0, 'canceled')
183 return
184 load_filename = str(load_filename)
185 load_dirname = load_filename + '_files'
186
187 if not os.path.isdir(load_dirname):
188 self.setProgress(0, 'directory not found')
189 print 'directory not found'
190 return
191
192 self.setProgress(50, 'clearing the face widgets')
193 self.face_widgets = []
194 for i in reversed(range(self.faces_layout.count())):
195 wid = self.faces_layout.itemAt(i).widget()
196 self.faces_layout.removeWidget(wid)
197 wid.setParent(None)
198
199 self.setProgress(75, 'loading')
200 with open(load_filename, 'r') as file:
201 for row in csv.reader(file, delimiter=' ', quotechar='"'):
202 name = row[0]
203 img_filename = load_dirname + '/' + row[1]
204 img = cv2.imread(img_filename)
205 if img is None:
206 print 'failed to open the image'
207 continue
208
209 #img = self.improve(img)
210 face_widget = FaceAndNameWidget(img, name)
211 self.face_widgets.append(face_widget)
212 self.faces_layout.addWidget(self.face_widgets[-1])
213 print name, img_filename
214 self.setProgress(100, 'done')
215
216 # callback for send button pushing event
217 # sends faces to a ROS node

Callers

nothing calls this directly

Calls 3

setProgressMethod · 0.95
FaceAndNameWidgetClass · 0.85
countMethod · 0.80

Tested by

no test coverage detected