(self)
| 59 | self._setup_cropper() |
| 60 | |
| 61 | def _setup_cropper(self): |
| 62 | self.Cropper = {} |
| 63 | for crop_part in ['head', 'hand']: |
| 64 | data_cfg = self.cfg.dataset[crop_part] |
| 65 | scale_size = (data_cfg.scale_min + data_cfg.scale_max)*0.5 |
| 66 | self.Cropper[crop_part] = tensor_cropper.Cropper( |
| 67 | crop_size=data_cfg.image_size, |
| 68 | scale=[scale_size, scale_size], |
| 69 | trans_scale=0) |
| 70 | |
| 71 | def _create_model(self): |
| 72 | self.model_dict = {} |