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)
可以把mask复制到原图目录下并验证一下
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
文章评论