python實(shí)現(xiàn)圖像隨機(jī)裁剪的示例代碼
import cv2 as cv import numpy as npnp.random.seed(0)# get IoU overlap ratiodef iou(a, b):# get area of a area_a = (a[2] - a[0]) * (a[3] - a[1])# get area of b area_b = (b[2] - b[0]) * (b[3] - b[1])# get left top x of IoU iou_x1 = np.maximum(a[0], b[0])# get left top y of IoU iou_y1 = np.maximum(a[1], b[1])# get right bottom of IoU iou_x2 = np.minimum(a[2], b[2])# get right bottom of IoU iou_y2 = np.minimum(a[3], b[3])# get width of IoU iou_w = iou_x2 - iou_x1# get height of IoU iou_h = iou_y2 - iou_y1# get area of IoU area_iou = iou_w * iou_h# get overlap ratio between IoU and all area iou = area_iou / (area_a + area_b - area_iou) return iou# crop and create databasedef crop_bbox(img, gt, Crop_N=200, L=60, th=0.5): # get shape H, W, C = img.shape # each crop for i in range(Crop_N): # get left top x of crop bounding box x1 = np.random.randint(W - L) # get left top y of crop bounding box y1 = np.random.randint(H - L) # get right bottom x of crop bounding box x2 = x1 + L # get right bottom y of crop bounding box y2 = y1 + L # crop bounding box crop = np.array((x1, y1, x2, y2)) # get IoU between crop box and gt _iou = iou(gt, crop) # assign label if _iou >= th: cv.rectangle(img, (x1, y1), (x2, y2), (0,0,255), 1) label = 1 else: cv.rectangle(img, (x1, y1), (x2, y2), (255,0,0), 1) label = 0 return img# read imageimg = cv.imread('../xiyi.jpg')img1 = img.copy()# gt bounding boxgt = np.array((87, 51, 169, 113), dtype=np.float32)# get crop bounding boximg = crop_bbox(img, gt, Crop_N=100, L=60, th=0.6)# draw gtcv.rectangle(img, (gt[0], gt[1]), (gt[2], gt[3]), (0,255,0), 1)cv.rectangle(img1,(gt[0], gt[1]), (gt[2], gt[3]), (0,255,0), 1)cv.imshow('result1',img1)cv.imshow('result', img)cv.imwrite('out.jpg', img)cv.waitKey(0)cv.destroyAllWindows()實(shí)驗(yàn)結(jié)果:
以上就是python實(shí)現(xiàn)圖像隨機(jī)裁剪的示例代碼的詳細(xì)內(nèi)容,更多關(guān)于python 圖像裁剪的資料請(qǐng)關(guān)注好吧啦網(wǎng)其它相關(guān)文章!
相關(guān)文章:
1. 利用FastReport傳遞圖片參數(shù)在報(bào)表上展示簽名信息的實(shí)現(xiàn)方法2. python讓函數(shù)不返回結(jié)果的方法3. jsp實(shí)現(xiàn)登錄驗(yàn)證的過(guò)濾器4. PHP循環(huán)與分支知識(shí)點(diǎn)梳理5. 低版本IE正常運(yùn)行HTML5+CSS3網(wǎng)站的3種解決方案6. Python TestSuite生成測(cè)試報(bào)告過(guò)程解析7. chat.asp聊天程序的編寫(xiě)方法8. 詳解JAVA 強(qiáng)引用9. python之cur.fetchall與cur.fetchone提取數(shù)據(jù)并統(tǒng)計(jì)處理操作10. JSP之表單提交get和post的區(qū)別詳解及實(shí)例
