Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Very slow prediction [object detection] #2503

Open
cadpm opened this issue Nov 17, 2024 · 4 comments
Open

Very slow prediction [object detection] #2503

cadpm opened this issue Nov 17, 2024 · 4 comments
Assignees
Labels
cv help wanted Extra attention is needed

Comments

@cadpm
Copy link

cadpm commented Nov 17, 2024

I am doing object detection with a pp-yolo-plus-S model.

And I'm getting prediction time of 300mS approximately.

Using a model trained by model yolov5-S I'm getting predictions of 90mS.

How can I reduce the prediction time?

I am generating my prediction with the following structure:

`
from paddlex import create_pipeline, create_model

model_deteccion = create_model("PaddleX/output01/best_model/inference", device="GPU")

frame="myimage.jpg"

myPrediction=list(model_deteccion.predict(frame)

`

@cuicheng01
Copy link
Collaborator

Could you provide a sample of your test image, and could you also provide the specifications of your GPU?

@cadpm
Copy link
Author

cadpm commented Nov 18, 2024

Sure, my GPU is rtx3060 12gb

@cadpm
Copy link
Author

cadpm commented Nov 18, 2024

The problem is not the image, it is the processing time.

I did Testing with PP-yolov-plus-s and yolov5-s both with coco models and evaluation with coco images. The prediction performance is 3 times less in Yolov5 (with pytorch)

@cuicheng01
Copy link
Collaborator

Could you kindly explain how your PyTorch model makes predictions? If it’s convenient, could you share some example code?

@AmberC0209 AmberC0209 added help wanted Extra attention is needed cv labels Nov 27, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cv help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

4 participants