yolo_slowfast 请问在哪里改batchsize大小

请问在哪里改batchsize大小

Open txyxju1997 opened this issue 2 years ago • 6 comments

Mar 23 '22 07:03 txyxju1997

同问，找到了吗？

May 06 '22 02:05 Bruce-Si

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

May 20 '22 00:05 wufan-tb

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

May 20 '22 07:05 Bruce-Si

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

May 20 '22 07:05 wufan-tb

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

May 20 '22 09:05 Bruce-Si