1 Star 0 Fork 272

AGUI/PaddleDetection

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
修改体.py 2.73 KB
一键复制 编辑 原始数据 按行查看 历史
AGUI 提交于 2024-03-20 06:56 . 1
for i in list(bbox_res):
print(i)
print(i['bbox'])
print([int(x) for x in i['bbox']])
li=[int(x) for x in i['bbox']]
# if li[0]>li[2] and li[1]>li[3]:
# li[0], li[2] = li[2], li[0]
# li[1], li[3] = li[3], li[1]
# if li[0]<li[2] and li[1]<li[3]:
# 将图片转换为NumPy数组
# image_array = np.array(zero_im)
image_array = np.array(image)
if len(li) == 4:
# draw bbox
xmin, ymin, w, h = li
xmax = xmin + w
ymax = ymin + h
# draw.line(
# [(xmin, ymin), (xmin, ymax), (xmax, ymax), (xmax, ymin),
# (xmin, ymin)],
# width=2,
# fill=color)
elif len(li) == 8:
x1, y1, x2, y2, x3, y3, x4, y4 = li
# draw.line(
# [(x1, y1), (x2, y2), (x3, y3), (x4, y4), (x1, y1)],
# width=2,
# fill=color)
xmin = min(x1, x2, x3, x4)
ymin = min(y1, y2, y3, y4)
xmax = max(x1, x2, x3, x4)
ymax = max(y1, y2, y3, y4)
else:
logger.error('the shape of bbox must be [M, 4] or [M, 8]!')
# 定义你的区域坐标
x1, y1, x2, y2 = xmin,ymin,xmax,ymax
# 将指定区域的像素值设置为True
region_matrix[y1:y2, x1:x2] = True
# 使用遮罩数组来设置图像数组的像素值
# 将布尔矩阵转换为uint8矩阵,以便与原始图片的像素值兼容
# region_matrix1 = region_matrix.astype(np.uint8) * 255
# 将修改后的数组转换回图片
# new_image = Image.fromarray(region_matrix1)
# 显示新的图片
image_array[region_matrix == False] = 0
result_image = Image.fromarray(image_array)
# 显示结果图像
result_image.show()
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/AGUIAtNight/PaddleDetection.git
git@gitee.com:AGUIAtNight/PaddleDetection.git
AGUIAtNight
PaddleDetection
PaddleDetection
release/2.6

搜索帮助

0d507c66 1850385 C8b1a773 1850385