yolo_test.py

# -*- coding: utf-8 -*-
“““
功能：keras-yolov3 进行批量测试 并 保存结果
项目来源：https://github.com/qqwweee/keras-yolo3
“““

import colorsys
import os
from timeit import default_timer as timer
import time

import numpy as np
from keras import backend as K
from keras.models import load_model
from keras.layers import Input
from PIL import Image ImageFont ImageDraw

from yolo3.model import yolo_eval yolo_body tiny_yolo_body
from yolo3.utils import letterbox_image
from keras.utils import multi_gpu_model

path = ‘./test_images/‘  #待检测图片的位置

# 创建创建一个存储检测结果的dir
result_path = ‘./out_images/‘
if not os.path.exists（result_path）:
    os.makedirs（result_path）

# result如果之前存放的有文件，全部清除
for i in os.listdir（result_path）:
    path_file = os.path.join（result_pathi）  
    if os.path.isfile（path_file）:
        os.remove（p