labelme点到图像外面导致崩溃的解决办法

最新推荐文章于 2024-07-01 19:20:57 发布

刀么克瑟拉莫

最新推荐文章于 2024-07-01 19:20:57 发布

阅读量164

点赞数

分类专栏：小功能 deeplearning 文章标签： labelme

本文链接：https://blog.youkuaiyun.com/random_repick/article/details/131333705

版权

deeplearning 同时被 2 个专栏收录

77 篇文章

订阅专栏

小功能

22 篇文章

订阅专栏

这段代码首先在图像四周填充黑色区域，然后读取并处理指定目录下的图片。接着，它将标注的黑白图截取回原图大小，并保存。最后，将mask复制到原图目录下并进行验证。整个过程涉及图像的读取、裁剪、颜色通道操作以及可视化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

先在图像四周填充黑色区域，这里图像高512，宽1280

import os, sys, cv2
import numpy as np

al= os.listdir('update')
al.sort()
for a in al:
	tmp = np.zeros((640, 1400,3)).astype('uint8')
	im = cv2.imread('update/'+a)
	tmp[50:562, 50:1330, :] = im
	cv2.imwrite('bigger/'+a, tmp)

标注并提取黑白图
从黑白图上截取原图区域

import os, sys, cv2, glob

path='bigger'
al=glob.glob(path+'/*json')
al.sort()
for a in al:
	if a[-5] != '.': continue
	os.system('labelme_json_to_dataset '+a)
	b = a.replace('.json', '_json')
	im = cv2.imread(b+'/label.png', 0)
	tmp=im[50:562, 50:1330]
	tmp[tmp>0] = 255
	name = a[:-5]+'.png'
	print(tmp.shape)
	cv2.imwrite(name, tmp)

import os, sys, cv2, glob
import numpy as np

path='update'
al=glob.glob(path+'/*jpg')
al.sort()
for a in al:
	b=a.replace('jpg', 'png')
	im = cv2.imread(a)
	mask = cv2.imread(b, 0)
	print(mask.shape)
	tmp = np.zeros(im.shape).astype('uint8')
	tmp[..., 2] = mask/2
	im[mask>0] = im[mask>0] / 2 + tmp[mask>0]
	cv2.imshow('ss', im)
	if cv2.waitKey() & 0xff == 27: break