search

XinzeLee / PolygonObjectDetection

This repository is based on Ultralytics/yolov5, with adjustments to enable polygon prediction boxes.
352 stars 95 forks source link

Inference with CPU #7

Open alvarezpardo opened 3 years ago

alvarezpardo commented 3 years ago

Hi! First of all, thanks for such a great work!

I would like to ask you if it is possible to perform the inference only with CPU, due to the GPU installation requirements, and if you know the RAM usage. I need to embedd a detection or semantic segmentation CNN and we have only CPU and very limited available memory (200Mb)

Thank you very much!

XinzeLee commented 3 years ago

Thank you for your interests in Polygon-Yolov5. For Polygon-Yolov5s model inferencing via GPU, the GPU memory taken is less than 1GB and is larger than 0.5GB. Definitely, you can do inference with CPU, since it is feasible (I have used my own laptop with CPU only to do the inference).

alvarezpardo commented 3 years ago

Thanks for the info!

Nifdi01 commented 1 year ago

@XinzeLee Can you give steps on how to use CPU instead of GPU? I am not sure on how to do the setup for CPU Thanks in advance