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

Custom engine model name not changing #18

Closed
amish25 opened this issue Dec 28, 2020 · 4 comments
Closed

Custom engine model name not changing #18

amish25 opened this issue Dec 28, 2020 · 4 comments

Comments

@amish25
Copy link

amish25 commented Dec 28, 2020

Hi, firstly thanks for your great work. Small issue, when I create an engine with a custom engine name using your native folder, the engine doesn't have the same name as that specified in the config file.

For example, if I set model-engine-file=model_b1_gpu0_fp32_custom.engine, the engine is saved as model_b1_gpu0_fp32.engine.

@marcoslucianops
Copy link
Owner

For example, if I set model-engine-file=model_b1_gpu0_fp32_custom.engine, the engine is saved as model_b1_gpu0_fp32.engine.

This is an implementation of DeepStream. You can change it only by editing and recompiling default DeepStream code (not Yolo model code). The easy way is rename file and edit config_infer path.

@gameliee
Copy link

gameliee commented Apr 1, 2022

This is an implementation of DeepStream.

Hi, could you provide the reference to this

@marcoslucianops
Copy link
Owner

@ntd94, refer to /opt/nvidia/deepstream/deepstream-6.0/sources/libs/nvdsinfer/nvdsinfer_model_builder.cpp lines 825-827:

suggestedPathName =
        modelPath + "_b" + std::to_string(initParams.maxBatchSize) + "_" +
        devId + "_" + networkMode2Str(networkMode) + ".engine";

@hupu1dong
Copy link

how to recompiling default DeepStream code,I change /opt/nvidia/deepstream/deepstream-6.0/sources/libs/nvdsinfer/nvdsinfer_model_builder.cpp @marcoslucianops

This issue was closed.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants