def see_big_img(): yolo = YOLO() with open('train.txt') as f: lines = f.readlines() np.random.shuffle(lines) for k in lines[:50]: line = k.split() img2 = draw(line) try: image = cv2.imread(line[0]) if image is None: raise FileNotFoundError('666') except Exception: print('Open Error! Try again!') continue else: image = yolo.detect_big_image(image) cv2.imshow('img', image) cv2.imshow('img1', img2) cv2.waitKey(0) cv2.destroyAllWindows() k = input('233') if k.lower() == 'q': break yolo.close_session()
def detect_img(): yolo = YOLO() while True: img = input('Input image filename(input Q to exit):') if img.lower() == 'q': yolo.close_session() return try: image = cv2.imread(img) if image is None: raise FileNotFoundError('666') except Exception: print('Open Error! Try again!') continue else: r_image = yolo.detect_big_image(image) cv2.imshow('img', r_image) cv2.waitKey(0) cv2.destroyWindow('img') yolo.close_session()