TensorRT-CenterNet
TensorRT-CenterNet copied to clipboard
Issue when applying mobileNet
Hi,i tried the mobilenetV2 you provided in the other repo.It successfully converted to the desired engine.But when i run the runDet command and evaluation command.It throws out the exception:
terminate called after throwing an instance of 'thrust::system::system_error' what(): parallel_for failed: no kernel image is available for execution on the device
I don't know where the problem came from.Anybody could help?
Btw the mobilenet setting as get_mobile_net(5,{'hm':18,'reg':2,'wh':2}),the head_conv set as default 64. I'm not sure if the head_conv number is too large or should i set the head_conv as 24