search

wufan-tb / yolo_slowfast

Yolov5+SlowFast: Realtime Action Detection Based on PytorchVideo
417 stars 52 forks source link

请问在哪里改batchsize大小 #3

Open txyxju1997 opened 2 years ago

Bruce-Si commented 2 years ago

同问，找到了吗？

wufan-tb commented 2 years ago

这个里面不涉及batchsize，因为slowfast是一秒的视频处理一次，所以这一秒的（比如25帧）就作为一个batch送到yolov5里处理，不方便修改，这也是现有框架下比较好的方式（基本实时），硬要改的话，可以看下slow_fast.py 里面第100~110行代码

Bruce-Si commented 2 years ago

@wufan-tb 明白了谢谢，但为啥我在v100上slowfast处理1个clip的视频要5s左右 1

wufan-tb commented 2 years ago

你这个好像是有点慢，确认gpu用上了吗，或者有可能和视频尺寸有关系，压缩下分辨率和码率呢

Bruce-Si commented 2 years ago

嗯，确定用上了gpu，视频刚压缩了一下，对slowfast速度没影响... 1

white1wind commented 1 year ago

你这个好像有点慢，确认gpu用上了吗，或者有可能和视频尺寸有关系，压缩率和码率呢

你好，我这为什么用不上gpu