如果測試圖像而非視頻的話,網上有很多代碼都是在terminal裏面運行:python yolo_video.py --image, 本人比較懶不喜歡使用命令運行程序,因此對測試程序yolo.video.py做了如下更改。
實驗結果發現Keras寫的yolo雖然可以出實驗結果但是loss接近10左右。實驗結果如圖:
將yolo_video.py更改爲下面的代碼直接運行即可,每測試一張顯示一張圖像,22行的path是測試圖像的路徑,23行outputdir是要保存測試圖像的路徑,61行的default=‘True’即可:(使用需要點贊,謝謝!)
import sys
import argparse
from yolo import YOLO, detect_video
from PIL import Image
import glob, os
from skimage import io
from matplotlib import pyplot as plt
import numpy as np
# def detect_img(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()
# yolo.close_session()
def detect_img(yolo):
path = "D:\\Users\\Experiments\\YOLO3\\keras-yolo3-master\\data\\testimage\\*.jpg"
outputdir = "D:\\Users\\Experiments\\YOLO3\\keras-yolo3-master\\data"
for jpgfile in glob.glob(path):
img = Image.open(jpgfile)
img = yolo.detect_image(img)
img.save(os.path.join(outputdir, os.path.basename(jpgfile)))
img = np.array(img)
io.imshow(img)
plt.show()
yolo.close_session()
FLAGS = None
if __name__ == '__main__':
# class YOLO defines the default value, so suppress any default here
parser = argparse.ArgumentParser(argument_default=argparse.SUPPRESS)
'''
Command line options
'''
parser.add_argument(
'--model', type=str,
help='path to model weight file, default ' + YOLO.get_defaults("model_path")
)
parser.add_argument(
'--anchors', type=str,
help='path to anchor definitions, default ' + YOLO.get_defaults("anchors_path")
)
parser.add_argument(
'--classes', type=str,
help='path to class definitions, default ' + YOLO.get_defaults("classes_path")
)
parser.add_argument(
'--gpu_num', type=int,
help='Number of GPU to use, default ' + str(YOLO.get_defaults("gpu_num"))
)
parser.add_argument(
'--image', default=True, action="store_true",
help='Image detection mode, will ignore all positional arguments'
)
'''
Command line positional arguments -- for video detection mode
'''
parser.add_argument(
"--input", nargs='?', type=str,required=False,default='./path2your_video',
help = "Video input path"
)
parser.add_argument(
"--output", nargs='?', type=str, default="",
help = "[Optional] Video output path"
)
FLAGS = parser.parse_args()
if FLAGS.image:
"""
Image detection mode, disregard any remaining command line arguments
"""
print("Image detection mode")
if "input" in FLAGS:
print(" Ignoring remaining command line arguments: " + FLAGS.input + "," + FLAGS.output)
detect_img(YOLO(**vars(FLAGS)))
elif "input" in FLAGS:
detect_video(YOLO(**vars(FLAGS)), FLAGS.input, FLAGS.output)
else:
print("Must specify at least video_input_path. See usage with --help.")