search

wufan-tb / yolo_slowfast

Yolov5+SlowFast: Realtime Action Detection Based on PytorchVideo
438 stars 55 forks source link

请问下如何设置slowfast动作检测的类别？ #7

Open xiebhty opened 2 years ago

xiebhty commented 2 years ago

请问下如何设置slowfast动作检测的类别？如何设置或者修改代码使得，slowfast只检测指定的某些动作，谢谢！

wufan-tb commented 2 years ago

这个目前还不支持设置修改呢，可以在yolo_slowfast.py代码的第71行下面增加一个判定条件，不是指定动作就不画图

if ava_label not in [action1, action2, ...]:
    continue

要稍微注意一下ava_label是字符串不是数字id

chutao-1 commented 11 months ago

@xiebhty 请问你解决了吗，求回复

chutao-1 commented 10 months ago

这个目前还不支持设置修改呢，可以在yolo_slowfast.py代码的第71行下面增加一个判定条件，不是指定动作就不画图

if ava_label not in [action1, action2, ...]:
    continue

要稍微注意一下ava_label是字符串不是数字id

@wufan-tb 请问这个是加在现在更新后版本代码的108行后面吗？我尝试了结果视频一个动作都不显示出来呀，还是原视频输出的，求回复

ChangXinze commented 8 months ago

@chutao-1 你加的代码是什么内容呢

chutao-1 commented 8 months ago

已经解决了，加的就博主说的 if ava_label not in 那个代码，就是效果不好

------------------ 原始邮件 ------------------ 发件人: "Xinze @.>; 发送时间: 2024年1月29日(星期一) 上午10:49 收件人: @.>; 抄送: @.>; @.>; 主题: Re: [wufan-tb/yolo_slowfast] 请问下如何设置slowfast动作检测的类别？ (Issue #7)

@chutao-1 你加的代码是什么内容呢

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>