1

I want to calculate and curve a graph for mAP/epochs for a training dataset. Here is my training code. I used this github source below to add in relevent code. But there were some errors because that was for YOLO. While I used faster r-cnn and code with pytorch for a pascal voc custom dataset.

Could you please help me how to add "IOU" and "mAP" calculation code to curve mAP/epochs graph??

mAP and IoU metrics

my training code:

num_epochs = 50
vector_row =[] 
for epoch in range(num_epochs):       
    loss = train_one_epoch(model, optimizer, train_loader)
    vector_row.append(loss) 
    print('epoch [{}]:  \t lr: {}  \t MSE: {}  '.format(epoch, lr_scheduler.get_last_lr(), loss))
    lr_scheduler.step()
marc_s
  • 732,580
  • 175
  • 1,330
  • 1,459

0 Answers0