search

bit-bots / YOEO

YouOnlyEncodeOnce - A CNN for Embedded Object Detection and Semantic Segmentation
GNU General Public License v3.0
21 stars 4 forks source link

fix: detect_image_cuda_tensors_to_cpu_missing #25

Closed phinik closed 2 years ago

phinik commented 2 years ago

fix: in detect_image from detect.py, cuda tensors were not moved back to the cpu. Therefore, conversion to numpy array was not possible.