simple-faster-rcnn-pytorch icon indicating copy to clipboard operation
simple-faster-rcnn-pytorch copied to clipboard

HI,How to achieve batch_size>1 or use multi_gpu,Can you give me an idea?

Open VikingKang opened this issue 7 years ago • 4 comments
trafficstars

VikingKang avatar Jan 13 '18 02:01 VikingKang

You need to revise dataloader(crop/resize/pad image to the same shape), RPN, Trainer etc. see faster-rcnn.pytorch for more detail

chenyuntc avatar Jan 13 '18 02:01 chenyuntc

Hello! Have you achieved that? I am still struggling!

CAROLZXYZXY avatar May 24 '19 17:05 CAROLZXYZXY

You need to revise dataloader(crop/resize/pad image to the same shape), RPN, Trainer etc. see faster-rcnn.pytorch for more detail

Hi! I wonder which parts exactly I need to look into if I want to change batch_size = 1 to batch_size = N in your code, you mentioned dataloader, RPN, trainer,etc. What this etc. include?

Thank you very much!

300LiterPropofol avatar Mar 19 '20 13:03 300LiterPropofol

@CAROLZXYZXY @300LiterPropofol Have you solved this problem?I'll be grateful if you can give me some suggestions!

ZhuYingJessica avatar Apr 11 '20 02:04 ZhuYingJessica