Skip to content

freenowill/Object-Detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

object-detection

利用tensorflow以及keras框架实现YOLO算法,对图像进行目标检测| yolo_filter_boxes:对YOLO boxes进行过滤| iou:计算交并比| yolo_non_max_suppression:对boxes进行非最大抑制(NMS)| yolo_eval:对yolo的编码进行处理转换(包括对boxes的过滤、非最大抑制等),返回scores,boxes,classes| predict:对自己的图片进行检测

yolo_utils

read_classes:读取对象变量(coco_classes.txt),返回包含识别对象字符串的列表| read_anchors:读取anchors(yolo_anchors.txt),返回anchors(n x 2)| generate_colors:对检测对象产生不同的颜色| preprocess_image:将输入图像处理为(608 x 608)| draw_boxes:绘制boxes| scale_boxes:将boxes进行尺度变换以适用于当前图片大小

yolo.h5

为节省训练时间,下载已经训练好的模型yolu.h5| 具体方法:https://blog.csdn.net/weixin_41043240/article/details/79745614?utm_source=blogxgwz8,或者下载我做好的yolu.h5:链接: 链接: https://pan.baidu.com/s/1czsFrhSAlK0cF7qvykZj9Q 提取码: ygfs

yad2k.models.keras_yolo.py

yolo_head:将模型输出转换为需要的形式| yolo_boxes_to_corners:将yolo boxes转换为(ymin,xmin,ymax,xmax)表示的边界

model_data

包含coco_classes.txt object_classes.txt yolo_anchors.txt

font

绘图所用字体

About

利用tensorflow以及keras框架实现YOLO算法,对图像进行目标检测

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages