custom-lib for img size 416

marcoslucianops / DeepStream-Yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models

MIT License

1.48k stars 357 forks source link

custom-lib for img size 416 #568

Open thewantedlegend opened 2 months ago

thewantedlegend commented 2 months ago

I compiled the code. But when I am trying to test model damoyolo_tinynasL20_Nl.onnx it exports an engine and gives error

terminate called after throwing an instance of 'thrust::system::system_error' what(): trivial_device_copy D->H failed: cudaErrorIllegalAddress: an illegal memory access was encountered Aborted (core dumped)

When I try with a model trained with size=640 it runs without any error.

marcoslucianops commented 1 week ago

Are you converting the pth file to onnx using the utils/export_damoyolo.py file?