The code currently only checks for the MPS backend which is for apple silicon only:
https://github.com/davepl/ObjectDetect/blob/main/detect.py#L20
You need to switch it to this to use your CUDA cores:
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
The code currently only checks for the MPS backend which is for apple silicon only:
https://github.com/davepl/ObjectDetect/blob/main/detect.py#L20
You need to switch it to this to use your CUDA cores:
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")