import json
import numpy as np
from labelme import utils
import cv2
import os
def create_file(file):
if not os.path.exists(file):
os.mkdir(file)
def cv_show(name,img):
cv2.imshow(name,img)
cv2.waitKey()
cv2.destroyAllWindows()
def from_mask_extract_ROI(img,mask):
y = np.where(mask==1)[0]
x = np.where(mask==1)[1]
lu_y = min(y)
ld_y = max(y)
ru_x = min(x)
rd_x = max(x)
mask_ROI = mask[lu_y:ld_y+1,ru_x:rd_x+1]
ROI = img[lu_y:ld_y+1,ru_x:rd_x+1]
mask = np.ones_like(ROI)
for i in range(mask.shape[2]):
mask[:,:,i] = mask_ROI
masked_ROI = mask*ROI
channel_reverse = np.ones_like(masked_ROI)
for i in range(channel_reverse.shape[2]):
channel_reverse[:,:,-1-i] = masked_ROI
labelme标准数据并抠图
最新推荐文章于 2022-12-07 00:19:44 发布