大家首先进入objdetector.py文件中
在最上面加入这段代码:
import torch.backends.cudnn as cudnn
接着在大约62行的地方,加入这段代码:
cudnn.benchmark = True
最后在demo.py文件下去运行代码,即可运行成功!
大家首先进入objdetector.py文件中
在最上面加入这段代码:
import torch.backends.cudnn as cudnn
接着在大约62行的地方,加入这段代码:
cudnn.benchmark = True
最后在demo.py文件下去运行代码,即可运行成功!