bingykang / Fewshot_Detection

Few-shot Object Detection via Feature Reweighting
https://arxiv.org/abs/1812.01866
529 stars 112 forks source link

ValueError: Expected input batch_size (20) to match target batch_size (8). #87

Open zhanjunyao opened 1 year ago

zhanjunyao commented 1 year ago

after I prepared data and trained base model as Readme,the error happended in region_loss.py loss_cls = self.class_scale * ClassificationLoss(cls, tcls) I don't know where goes error,someone has ever met this problem?how to solve? thank you