Support yolov8 pose estimation

[tonhathuy]

Thank you, this project is great!
can I run this repo with yolov8 pose estimation? What do I have to adjust in the code to make it work?
thanks

[jveitchmichaelis]

If you can export your model as an EdgeTPU file then you would need to add the post-processing code to convert the output into whatever format you expect the keypoints to be in. So the first step would be to run the model and inspect the output - if it's just a bunch of keypoints then you might not need to do much else. I guess minimally you have to re-scale the coordinates to the original image dimensions?

Currently there is no "support" for pose estimation, in that the repo was built for object detection (so there are a bunch of other post-processing steps like non-max suppression, coordinate scaling, etc). However the core code that takes an image and runs it through the EdgeTPU should work with your model. I'd be happy to merge a PR if you get it working and feel free to ask if you have questions about the code.

e.g. this function:

edgetpu-yolo/edgetpumodel.py

Line 135 in 283644d

def forward(self, x:np.ndarray, with_nms=True) -> np.ndarray: