search

ArgoHA / general_triton_yolo_pipeline

This is a repo with a Triton Server deployment template
19 stars 4 forks source link

Will this pipeline work for YoloV7 segmentation models too? #2

Open RishabhSheoran opened 1 month ago

RishabhSheoran commented 1 month ago

Hey Argo! can this pipeline also be used for Yolo V7 segmentation model?

ArgoHA commented 1 month ago

Hey @RishabhSheoran, I didn't test it unfortunately. Main question here is if yolov7 preprocessing and postprocessing is the same as in yolov8. Input should be the same, but model's output might have a different shape. If you are seeing a specific issue - share the details, I'll try helping you make changes, so it runs.