search

Owen-Liuyuxuan / visualDet3D

Official Repo for Ground-aware Monocular 3D Object Detection for Autonomous Driving / YOLOStereo3D: A Step Back to 2D for Efficient Stereo 3D Detection
https://owen-liuyuxuan.github.io/papers_reading_sharing.github.io/3dDetection/GroundAwareConvultion/
Apache License 2.0
361 stars 76 forks source link

cpu #68

Open VictorLee0214 opened 1 year ago

VictorLee0214 commented 1 year ago

sorry mate，is there any way that i am able to work this project under the cpu？

Owen-Liuyuxuan commented 1 year ago

This repo is not very friendly to non-cuda user.

You can manually substitute all the ".cuda()" to ".cpu()". And clean up the training script could probably get you started.