-
-
Notifications
You must be signed in to change notification settings - Fork 253
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
Custom Detector (YoloX, YoloR, CentreDet) Integration #186
Comments
The general workflow is correct. Also it is recommended to use the same ONNX -> TensorRT conversion. You also want to create yolox.py or yolor.py for ONNX to TRT conversion. If extra TRT plugins are required, they need to be accounted for as well. What is YOLOX’s advantages over Scaled-YOLOv4? |
The usual claims of being better, faster, stronger haha https://towardsdatascience.com/how-to-train-yolox-on-a-custom-dataset-bb2f94cdb038 I'm just curious to try out and compare several different detectors. |
Performance looks similar to yolov4-csp-swish but it might be worth a try. You can try using torch2trt in yolox.py if it doesn’t add too many dependencies. |
Hey again @GeekAlexis
Great work on the repo so far. I would like to contribute by adding a few custom models like centrenet, yolor, yolox.
I know that these models need to be converted to ONNX or TensorRT first, perhaps using torch2trt. Could you please provide some general steps as to how to achieve this? From my understanding we need to do the following:
Thanks! I'll probably start with YoloX which already has TensorRT support
The text was updated successfully, but these errors were encountered: