xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-24 05:55:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add ncnn auto thread count (#3847)

Browse Source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
nihui 2023-07-21 07:38:59 +08:00 committed by GitHub
parent b251b7415a
commit 01dcd54b19
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/nn/autobackend.py
View File

@ -3,7 +3,6 @@
import ast import ast
import contextlib import contextlib
import json import json
import os
import platform import platform
import zipfile import zipfile
from collections import OrderedDict, namedtuple from collections import OrderedDict, namedtuple
@ -267,7 +266,6 @@ class AutoBackend(nn.Module):
check_requirements('git+https://github.com/Tencent/ncnn.git' if ARM64 else 'ncnn') # requires ncnn check_requirements('git+https://github.com/Tencent/ncnn.git' if ARM64 else 'ncnn') # requires ncnn
import ncnn as pyncnn import ncnn as pyncnn
net = pyncnn.Net() net = pyncnn.Net()
net.opt.num_threads = os.cpu_count()
net.opt.use_vulkan_compute = cuda net.opt.use_vulkan_compute = cuda
w = Path(w) w = Path(w)
if not w.is_file(): # if not *.param if not w.is_file(): # if not *.param