Exemplo n.º 1
0
# -*- 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()
Exemplo n.º 2
0
# -*- 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()
Exemplo n.º 3
0
# @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]
Exemplo n.º 4
0
# 导入包
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()
Exemplo n.º 5
0
# -*- 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()
Exemplo n.º 6
0
# -*- 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()