- Update 2020.7.16 增加deepsort，并作了大量调整
YOLOV3及其训练的实现借鉴：qqwweee/keras-yolo3
SORT的实现借鉴：abewley/sort
DeepSort的实现借鉴：theAIGuysCode/yolov3_deepsort
参考文献：
-
打开
yolo_video.py -
修改
DEFAULTS（个人原因不太喜欢用argparse）
DEFAULTS = {
"model_path": './model_h5/yolo.h5',
"anchors_path": './model_data/yolo_anchors.txt',
"classes_path": './model_data/coco_classes.txt',
"deepsort_model": './model_data/mars-small128.pb',
"gpu_num": 1,
"image": False, # 如果此处设置了True，"tracker"则被忽略
"tracker": 'deepsort', # 此处根据需要为'sort'或'deepsort'
"write_to_file": True,
"input": './input/your_video.format',
"output": './output/your_video.format',
"output_path": './output/',
"score": 0.4, # threshold
"iou": 0.4, # threshold
"repeat_iou": 0.95, # threshold
}
- 运行
yolo_video.py，结果可在"output_path"中指定的文件夹查看
python yolo_video.py
- 如果想适用轻量级的YOLOv3模型，修改'"model_path"'和'"anchors_path"'即可
*关于YOLOV3的内容，可以查看YOLO WEBSITE
*tiny-YOLOv3下载：tiny-YOLOv3
*YOLOv3下载：YOLOv3
*预训练的DeepSort网络：Google Drive: DeepSort, BaiduDisk: DeepSort pw:9i6p
model_path # h5文件路径
anchors_path # anchor的路径
classes_path # 存放识别对象类别的路径
deepsort_model # DeepSort预训练权重存放路径
gpu_num # gpu数
image # 处理video(False)或处理图片(True)
tracker # 是否使用追踪
write_to_file # 是否写入到文件
input # video的路径
output # 输出video的路径
output_path # 其他文件output的路径
score # 分数低于该阈值的物体会被忽略
iou # iou低于该阈值的物体会被忽略
repeat_iou # 去除重复bounding box
*写入到文件的格式为：
<frame>, <id>, <bb_left>, <bb_top>, <bb_width>, <bb_height>, <conf>, <x>, <y>, <z>
选取的图片从CIFAR-100 dataset中提取，由于主要研究对象是交通方面的，因此选取的物体种类主要围绕车辆和
人，详细分类见model_data/cifar_classes.txt
CIFAR数据集可在此网站查看：The CIFAR-10 and CIFAR-100
CIFAR-100 dataset下载：CIFAR-100 python version
- 可通过修改
read_data_cifar100.py确定训练数据集的物体类别
REMAIN = list(np.concatenate([[11, 35, 46, 98], [8, 13, 48, 58], [81, 85]]))
- 运行train.py
python train.py
可自行修改epochs，batch_size
- 可先使用训练好的YOLOv3模型
yolo.h5获取bounding box数据，再使用kmeans_anchors.py计算获得anchors
- 环境 Environment
-
主要依赖
- python 3.6
- Keras 2.3.1
- tensorflow-gpu 1.13.0
- numpy 1.17.0
(较低版本貌似也支持)
-
缺少
openh264-1.8.0-win64.dll可能会发生未知错误，因此需要将此文件和python yolo_video.py放置在 同一目录下（貌似少了也没啥事） -
DeepSort能解决短时遮挡问题，解决不了长时间object消失或被遮挡问题
-
DEMO上传至百度云 PW: pb34
-
MOT_DEMO Multiple Object Tracking Benchmark