技術標籤：影象視覺影象識別

#
#-------------------------------------#
#       對單張圖片進行預測
#-------------------------------------#
# from yolo import YOLO
# from PIL import Image

# yolo = YOLO()

# while True:
#     img = input('Input image filename:')
#     try:
#         image = Image.open(img)
#     except:
#         print('Open Error! Try again!')
 

#         continue
#     else:
#         r_image = yolo.detect_image(image)
#         r_image.show()
#對多張圖片進行檢測

from yolo import YOLO
from PIL import Image
import os

yolo = YOLO()
filepath ="/diskb/jyz/ddata/bus2/"
filename = os.listdir(filepath)
while True:
    for i in filename:
        print
 
(i)
        image = i.strip().split(".")
        j=i.split(".")[0]
        # print(image)
        # print(j)
#這條語句完全是為了練習split方法。

    # img = input('Input image filename:')
        try:
            img = image[0]+".jpg"
            imag = os.path.join(filepath,img)
            im =
 
 Image.open(imag)
            print(im)
        except:
            
            print('Open Error! Try again!')
            continue
        else:
            r_image = yolo.detect_image(im)
            r_image.save('/diskb/jyz/ddata/bus1/'+ j +'.jpg')
        break