YOLO-Patch-Based-Inference icon indicating copy to clipboard operation
YOLO-Patch-Based-Inference copied to clipboard

Is there way to support yolo-obb model?

Open filot977 opened this issue 1 year ago • 1 comments

Dear authors.

Do you have any plan to support yolo-obb model?

Thank you.

filot977 avatar Oct 01 '24 20:10 filot977

Good afternoon. Development in this area is underway. Most likely, a separate library called patched_obb_infer from our development team will be released in the next month. So stay tuned for updates. I will let you know when the first release is available. The current estimate is for early November.

Koldim2001 avatar Oct 01 '24 21:10 Koldim2001

Is there away to validate model output while using patch_inference as it usually boost the overall performance ?

previously I was able to do so with a tool like sahi inference after transforming annotations to coco annotations.

Alisoltan82 avatar Nov 06 '24 08:11 Alisoltan82

@Alisoltan82 Unfortunately, there isn't a ready-made tool to directly export patch inference results to COCO format. However, you can still obtain the results of the patch inference algorithm in a structured way, which you can then convert to any desired format, including COCO. Here's how you can access the results:

# Final Results:
img = result.image
confidences = result.filtered_confidences
boxes = result.filtered_boxes
polygons = result.filtered_polygons
classes_ids = result.filtered_classes_id
classes_names = result.filtered_classes_names

With these results, you can write the data into any format you need, including COCO. While we don't currently have a built-in tool for directly saving to COCO, we might consider adding such a feature in the future. For now, you can manually convert the results to COCO format using the data provided.

Koldim2001 avatar Nov 06 '24 19:11 Koldim2001

Thanks

Alisoltan82 avatar Nov 10 '24 21:11 Alisoltan82

Unfortunately, I haven't had time to implement this option yet. I'll let you know if this feature will appear in the next updates

Koldim2001 avatar Feb 09 '25 09:02 Koldim2001