deploy_ssd.py
5.69 KB
-
[BugFix] SSD fully supported on GPUs, updated deploy_ssd tutorial (#2510) · 48c16a17
* nms fixed for gpu, tested on cuda and opencl devices, ssd now can run fully on the gpu * sort updated to use virtual thread * typo fixed * fix lint * fix lint * add support when batch_size > 1 * intel graphics conv2d bugs fixed for inception_v3 * intel conv2d api updated, nn input size 4 condition added * review addressed * move conv_tags to attributes * opencl ctx fixed * nms_ir index simplified
Leyuan Wang committed