Skip to content

Commit 4665892

Browse files
committed
update device类型统一使用ascend
1 parent 4ad8b13 commit 4665892

3 files changed

Lines changed: 8 additions & 7 deletions

File tree

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
## OpencvToolsV1.1.4
1+
## OpencvToolsV1.1.5
22
Opencv相关操作
33

44
### 打包为wheel文件
@@ -17,3 +17,4 @@ pip wheel --wheel-dir=./wheel_dir ./
1717
* 支持华为Ascend解码
1818
* 日志输出Ascend芯片解码
1919
* 优化Ascend芯片解码
20+
* device类型统一使用ascend

opencv_tools/jade_opencv_process.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -763,7 +763,7 @@ def download_frame(self,frame):
763763
if self.use_gpu_decode:
764764
frame = frame.download()
765765
frame = cv2.cvtColor(frame, cv2.COLOR_BGRA2BGR)
766-
if self.device == "Ascend":
766+
if self.device == "ascend":
767767
frame = self._dvpp.jpege(frame)
768768
frame = frame.jpeg_to_cv2()
769769
return frame
@@ -823,7 +823,7 @@ def open_gpu_capture(self):
823823
Exit(0)
824824

825825
def opencv_cpu_capture(self):
826-
if self.device == "Ascend":
826+
if self.device == "ascend":
827827
from acllite import videocapture
828828
import acl
829829
ret = acl.rt.set_device(0)
@@ -833,7 +833,7 @@ def opencv_cpu_capture(self):
833833
self.capture = cv2.VideoCapture(self.video_path)
834834
self.JadeLog.INFO("相机类型为:{},使用CPU解码,准备打开相机".format(self.camera_type))
835835

836-
if self.device == "Ascend":
836+
if self.device == "ascend":
837837
ret,frame = self.capture.read()
838838
if ret == 0 and frame is not None:
839839
self.reopen_times = 0
@@ -877,7 +877,7 @@ def capture_reader(self):
877877
ret, frame = self.capture.nextFrame()
878878
else:
879879
ret, frame = self.capture.read()
880-
if self.device == "Ascend":
880+
if self.device == "ascend":
881881
if ret == 0:
882882
ret = True
883883
else:
@@ -902,7 +902,7 @@ def capture_reader(self):
902902
time.sleep(self.camera_reopen_times)
903903

904904
def run(self):
905-
if self.device == "Ascend":
905+
if self.device == "ascend":
906906
import acl
907907
from acllite.acllite_imageproc import AclLiteImageProc
908908
ret = acl.rt.set_device(0)

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
find_packages("opencv_tools", pack_list)
1313
setup(
1414
name="opencv_tools",
15-
version="1.1.4",
15+
version="1.1.5",
1616
keywords=("pip", "opencv_tools", ""),
1717
description="opencv_tools",
1818
long_description="",

0 commit comments

Comments
 (0)