实例使用:
importalbumentationsasA importrandom BOX_COLOR=(255,0,0) TEXT_COLOR=(255,255,255) defvisualize_bbox(img,bbox,color=BOX_COLOR,thickness=2,**kwargs): #height,width=img.shape[:2] x_min,y_min,w,h=bbox x_min,x_max,y_min,y_max=int(x_min),int(x_min+w),int(y_min),int(y_min+h) cv2.rectangle(img,(x_min,y_min),(x_max,y_max),color=color,thickness=thickness) returnimg
在图片处理中使用Albumentations操作图书馆仍然非常简单实用。如果你感兴趣,你也可以自己试试~