Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
scikit-image
Commits
ae7c2569
Commit
ae7c2569
authored
Aug 15, 2024
by
wangsen
Browse files
add test.py and readme.md
parents
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
77 additions
and
0 deletions
+77
-0
README.md
README.md
+50
-0
test.py
test.py
+27
-0
No files found.
README.md
0 → 100644
View file @
ae7c2569
# 下载DCU支持的cupy
```
wget https://cancon.hpccube.com:65024/directlink/1/DTK-24.04.1_apps-20240611/NFS3.2_CentOS7.6/cupy_v12.0.0b3_py38_nfs3.2_DTK24.04_29Mar2024.tar.gz
tar -zxvf cupy_v12.0.0b3_py38_nfs3.2_DTK24.04_29Mar2024.tar.gz
```
# 建立conda 环境
```
conda create -n scikit python=3.8
pip install cupy_v12.0.0b3_py38_nfs3.2_DTK24.04_29Mar2024/dist-py38/cupy-12.0.0b3-cp38-cp38-linux_x86_64.whl
pip install scikit-image -i https://pypi.tuna.tsinghua.edu.cn/simple
```
# 执行测试
```
import cupy as cp
import numpy as np
from skimage import data
from skimage import filters
from skimage import img_as_float
# 加载示例图像
image = img_as_float(data.coins())
# 将图像从NumPy数组转换为CuPy数组,以利用GPU
image_gpu = cp.array(image)
# 定义一个使用GPU的函数(例如,应用高斯滤波器)
def gaussian_filter_gpu(image, sigma=1):
from cupyx.scipy.ndimage import gaussian_filter
return gaussian_filter(image, sigma)
# 应用高斯滤波器
filtered_image_gpu = gaussian_filter_gpu(image_gpu, sigma=1)
# 将结果转换回NumPy数组
filtered_image = cp.asnumpy(filtered_image_gpu)
# 测试输出
print("Original Image Shape:", image.shape)
print("Filtered Image Shape:", filtered_image.shape)
```
\ No newline at end of file
test.py
0 → 100644
View file @
ae7c2569
import
cupy
as
cp
import
numpy
as
np
from
skimage
import
data
from
skimage
import
filters
from
skimage
import
img_as_float
# 加载示例图像
image
=
img_as_float
(
data
.
coins
())
# 将图像从NumPy数组转换为CuPy数组,以利用GPU
image_gpu
=
cp
.
array
(
image
)
# 定义一个使用GPU的函数(例如,应用高斯滤波器)
def
gaussian_filter_gpu
(
image
,
sigma
=
1
):
from
cupyx.scipy.ndimage
import
gaussian_filter
return
gaussian_filter
(
image
,
sigma
)
# 应用高斯滤波器
filtered_image_gpu
=
gaussian_filter_gpu
(
image_gpu
,
sigma
=
1
)
# 将结果转换回NumPy数组
filtered_image
=
cp
.
asnumpy
(
filtered_image_gpu
)
# 测试输出
print
(
"Original Image Shape:"
,
image
.
shape
)
print
(
"Filtered Image Shape:"
,
filtered_image
.
shape
)
\ No newline at end of file
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment