(self)
| 31 | logger.info(" Video Reid loaded.") |
| 32 | |
| 33 | def exec(self): |
| 34 | envelope = self.inp.recv() |
| 35 | if envelope is None: |
| 36 | return |
| 37 | msg = envelope.msg |
| 38 | |
| 39 | # for crop in image['shaper']: |
| 40 | # cv2.imwrite(f'reid_video_{envelope.partial_id}.jpg', crop) |
| 41 | # logger.info(f'envelope id {envelope.partial_id}') |
| 42 | |
| 43 | msg['features'] = [] |
| 44 | if 'crop' in msg: |
| 45 | msg['feature'] = self._model.inference(msg['crop']) |
| 46 | # logger.info(image['features']) |
| 47 | self.out.send(envelope) |