def cut_rects(img, rects): global img_idx label_img = op.get_subimage(img, rects[0][0],rects[0][1]) if not op.same_img(sub_imgs[0], label_img): return for i, rect in enumerate(rects): if i==0: continue img2 = op.get_subimage(img, rect[0], rect[1]) # dist = mc.diff(sub_imgs[i], img2) # if mc.average(dist) > 20: imgs = nc.get_single_numbers(img2) for number in imgs: cv2.imwrite('data/train5/'+str(img_idx)+'.jpg', number) img_idx += 1
def cut_rects(img, rects): global img_idx label_img = op.get_subimage(img, rects[0][0], rects[0][1]) if not op.same_img(sub_imgs[0], label_img): return for i, rect in enumerate(rects): if i == 0: continue img2 = op.get_subimage(img, rect[0], rect[1]) # dist = mc.diff(sub_imgs[i], img2) # if mc.average(dist) > 20: imgs = nc.get_single_numbers(img2) for number in imgs: cv2.imwrite('data/train5/' + str(img_idx) + '.jpg', number) img_idx += 1
def on_mouse(event, x, y, flags, param): global point_s global img global rects green = [0x0, 0xFF, 0x0] read = [0x0, 0x0, 0xFF] if event == cv2.cv.CV_EVENT_LBUTTONDOWN: img_cpy = img.copy() # copy image to draw rectangle point_s = (x, y) cv2.circle(img_cpy, point_s, 1, green) cv2.imshow(win_name, img_cpy) elif event == cv2.cv.CV_EVENT_RBUTTONDOWN: print 'cancel last region' if len(rects) != 0: rects.pop() sub_imgs.pop() elif event == cv2.cv.CV_EVENT_MOUSEMOVE and ( flags & cv2.cv.CV_EVENT_FLAG_LBUTTON): img_cpy = img.copy() # copy image to draw rectangle point_e = (x, y) if point_s != point_e: cv2.circle(img_cpy, point_e, 2, green) # draw point cv2.rectangle(img_cpy, point_s, point_e, read) cv2.imshow(win_name, img_cpy) elif event == cv2.cv.CV_EVENT_LBUTTONUP: point_e = (x, y) img_cpy = img.copy() # copy image to draw rectangle if point_s != point_e: cv2.circle(img_cpy, point_e, 2, green) cv2.rectangle(img_cpy, point_s, point_e, read) cv2.imshow(win_name, img_cpy) img2 = op.get_subimage(img, point_s, point_e) # img2 = cv2.resize(img2,(26,26)) # cv2.imwrite('heihei.jpg',img2) rects.append((point_s, point_e)) sub_imgs.append(img2) print 'add region', point_s, point_e
def on_mouse(event, x, y, flags, param): global point_s global img global rects green = [0x0,0xFF,0x0] read = [0x0,0x0,0xFF] if event == cv2.cv.CV_EVENT_LBUTTONDOWN: img_cpy = img.copy() # copy image to draw rectangle point_s = (x,y) cv2.circle(img_cpy, point_s, 1, green) cv2.imshow(win_name,img_cpy) elif event == cv2.cv.CV_EVENT_RBUTTONDOWN: print 'cancel last region' if len(rects) != 0: rects.pop() sub_imgs.pop() elif event == cv2.cv.CV_EVENT_MOUSEMOVE and (flags & cv2.cv.CV_EVENT_FLAG_LBUTTON): img_cpy = img.copy() # copy image to draw rectangle point_e = (x,y) if point_s != point_e: cv2.circle(img_cpy, point_e, 2, green) # draw point cv2.rectangle(img_cpy, point_s, point_e, read) cv2.imshow(win_name,img_cpy) elif event == cv2.cv.CV_EVENT_LBUTTONUP: point_e = (x,y) img_cpy = img.copy() # copy image to draw rectangle if point_s != point_e: cv2.circle(img_cpy, point_e, 2, green) cv2.rectangle(img_cpy, point_s, point_e, read) cv2.imshow(win_name,img_cpy) img2 = op.get_subimage(img, point_s, point_e) # img2 = cv2.resize(img2,(26,26)) # cv2.imwrite('heihei.jpg',img2) rects.append((point_s,point_e)) sub_imgs.append(img2) print 'add region', point_s, point_e