# -*- coding: utf-8 -*- # @Time : 2020/11/12 22:08 # @Author : AaronJny # @File : train_detect.py # @Desc : # 导入包 from xyolo import YOLO from xyolo import init_yolo_v3 from config import XYoloConfig # 使用修改后的配置创建yolo对象 config = XYoloConfig() init_yolo_v3(config) # 如果是训练，在创建yolo对象时要传递参数train=True yolo = YOLO(config, train=True) # 开始训练，训练完成后会自动保存 yolo.fit()
# -*- coding: utf-8 -*-
# @Time : 2020/10/31 15:20
# @Author : AaronJny
# @File : use_proxies.py
# @Desc : 使用预训练权重对图片执行目标检测
# 导入包
# 导入包
from xyolo import YOLO, DefaultYolo3Config
from xyolo import init_yolo_v3
# 创建默认配置类对象
config = DefaultYolo3Config()
# 初始化xyolo（下载预训练权重、转换权重等操作都是在这里完成的）
# 下载和转换只在第一次调用的时候进行，之后再调用会使用缓存的文件
init_yolo_v3(config)
# 创建一个yolo对象，这个对象提供使用yolov3进行检测和训练的接口
yolo = YOLO(config)
# 检测并在图片上标注出物体
img = yolo.detect_and_draw_image('./xyolo_data/detect.jpg')
# 展示标注后图片
img.show()
# @File : solution.py
# @Desc : 点选式验证码解决方案
import random
from glob import glob
import cv2
import numpy as np
from PIL import Image
from xyolo import YOLO
from classify_model import load_classify_model
from config import XYoloConfig
from process_classify_dataset import preprocess_image
config = XYoloConfig()
yolo = YOLO(config)
siamese_network = load_classify_model()
def detect(image_path):
# 接收图片
img = Image.open(image_path)
img_array = np.asarray(img.convert('RGB'))
# 提取两个标签
word1_array = img_array[1:34, 186:216]
word1_image = Image.fromarray(word1_array)
word2_array = img_array[1:34, 216:246]
word2_image = Image.fromarray(word2_array)
# 检测文字
results = yolo.detect_image(img)
word_images = [word1_image, word2_image]
# 导入包
from xyolo import YOLO, DefaultYolo3Config
from xyolo import init_yolo_v3
# 创建默认配置类对象
config = DefaultYolo3Config()
# 初始化xyolo（下载预训练权重、转换权重等操作都是在这里完成的）
# 下载和转换只在第一次调用的时候进行，之后再调用会使用缓存的文件
init_yolo_v3(config)
# 创建一个yolo对象，这个对象提供使用yolov3进行检测和训练的接口
yolo = YOLO(config)
# 检测并在图片上标注出物体
img = yolo.detect_and_draw_image('aaa.jpg')
# 展示标注后图片
img.show()
# -*- coding: utf-8 -*-
# @Time : 2020/11/12 22:28
# @Author : AaronJny
# @File : test_detect.py
# @Desc :
from xyolo import YOLO
from config import XYoloConfig
yolo = YOLO(XYoloConfig())
img = yolo.detect_and_draw_image('./images/286.png')
img.show()
# -*- coding: utf-8 -*-
# @Time : 2020/10/31 19:49
# @Author : AaronJny
# @File : eval_mydata.py
# @Desc :
from xyolo import DefaultYolo3Config
from xyolo import YOLO
class MyConfig(DefaultYolo3Config):
def __init__(self):
super(MyConfig, self).__init__()
self._classes_path = '/Users/aaron/code/xyolo/tests/xyolo_data/classes.txt'
self._model_path = '/Users/aaron/code/xyolo/tests/xyolo_data/output_model.h5'
config = MyConfig()
yolo = YOLO(config)
image_path = '/Users/aaron/code/bctt/spider/captcha_detection/soopat/images/232.png'
img = yolo.detect_and_draw_image(image_path)
img.show()