search

owo12321 / Yolov5_DeepSort_Traffic-counter

基于Yolov5_DeepSort的物体计数器，可以统计车流或人流量等
157 stars 26 forks source link

请问用的是什么数据集训练的 #27

Open sinpne opened 4 months ago

owo12321 commented 4 months ago

默认使用yolov5官方在coco数据集上训练出来的预训练权重，要训练的话参考https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data

owo12321 commented 4 months ago

deepsort使用的也是官方的权重，是在Market-1501上训练的