search

alanlukezic / d3s

D3S - Discriminative Single Shot Segmentation Tracker (CVPR 2020)
265 stars 56 forks source link

Can this repo run on MacBook? Without GPU. #10

Closed shoutOutYangJie closed 4 years ago

shoutOutYangJie commented 4 years ago

I set ATOM's "use_gpu" as False, but I meet "TypeError: __cuda_array_interface__ must be a dict"

alanlukezic commented 4 years ago

I have not tested the code on the Mac without a GPU. After you set the use_gpu to false you need to modify a few line of code in the pytracking/tracker/segm/segm.py: search for the "if self.params.use_gpu" in this file and add similar body in the "else" part of the condition.