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

can I set annother value of batchsize besides 1?

Open iwanli opened this issue 6 years ago • 3 comments
trafficstars

I want to set a bigger value of batch size,but it can not work when i change the value,it seems the value can only be 1.How can i set a bigger value? thank you!

iwanli avatar Nov 26 '18 07:11 iwanli

@chenyuntc Hello, I would also love to know this.

howardyclo avatar Dec 04 '18 16:12 howardyclo

@iwanli @howardyclo Hello, have you solved this problem? I'll be grateful if you can give me some suggestions

ZhuYingJessica avatar Apr 11 '20 02:04 ZhuYingJessica

it will take some works if u wanna train with batch size > 1. First, u should change the code for data processing, cuz each image has different number of objects, the ground truth label cannot be form a batch data directly, u can fill some 'non-objects' label in a batch, which is like the YOLO do; Second, u should change the code in the training part, mainly in the forward() function in class 'FasterRCNNTrainer'; Then, u can test your code and get more details.

chrisway613 avatar Jul 23 '20 10:07 chrisway613