xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-07-08 06:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Pose export from YAML (#3312)

Browse Source
This commit is contained in:
Glenn Jocher 2023-06-21 18:04:22 +02:00 committed by GitHub
parent 9d1e5567de
commit 35c02c8319
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/yolo/engine/exporter.py
View File

@ -224,7 +224,7 @@ class Exporter:
'imgsz': self.imgsz, 'imgsz': self.imgsz,
'names': model.names} # model metadata 'names': model.names} # model metadata
if model.task == 'pose': if model.task == 'pose':
self.metadata['kpt_shape'] = model.kpt_shape self.metadata['kpt_shape'] = model.model[-1].kpt_shape
LOGGER.info(f"\n{colorstr('PyTorch:')} starting from {file} with input shape {tuple(im.shape)} BCHW and " LOGGER.info(f"\n{colorstr('PyTorch:')} starting from {file} with input shape {tuple(im.shape)} BCHW and "
f'output shape(s) {self.output_shape} ({file_size(file):.1f} MB)') f'output shape(s) {self.output_shape} ({file_size(file):.1f} MB)')