search

pinery-sls / Pytorch_Yolor_DeepSort

Yolor and DeepSort
10 stars 4 forks source link

如何使用自己的yolor模型 #5

Open axonEmily opened 1 year ago

axonEmily commented 1 year ago

你好，我能成功運行默認的模型，但是我想用自己的yolor模型，請問要做哪些更改？我嘗試直接在命令行運行python track.py --source video_1.mp4 --show-vid --save-vid --yolo_weight yolor/weights/best_overall.pt，但是出現了錯誤信息AttributeError: 'collections.OrderedDict' object has no attribute 'float'