-
Notifications
You must be signed in to change notification settings - Fork 6.9k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
I am not able to obtain results with custom backbone #3040
Labels
Comments
I've got the same problem as well. I'll try to provide more information/context to maybe help us solve it.
Here is how my final model looks like:
let's train it for 10 epochs
As a result I am getting following output:
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
_I am following the tutorial about FasterRCNN and I would like to test my network as backbone of the net:
UCapsNet return 512 features maps
I am training on VocPascal 2007_
FRCN_model = FasterRCNN(backbone_model.Ucapsnet, 21, rpn_anchor_generator=backbone_model.anchor_generator, box_roi_pool=backbone_model.roi_pooler)
FRCN_model = FRCN_model.to(device)
params = [p for p in FRCN_model.parameters() if p.requires_grad]
optimizer = torch.optim.SGD(params, lr=0.02, momentum=0.9, weight_decay=1e-4)
pbar = tqdm(range(n_epochs))
for epoch in pbar:
train_one_epoch(FRCN_model, optimizer, dataloaders['train'], device, epoch, print_freq=10)
evaluate(FRCN_model, dataloaders['val'], device=device)
I got:
Averaged stats: model_time: 1605886336.0000 (1605886304.8101) evaluator_time: 0.0275 (0.0285)
Accumulating evaluation results...
DONE (t=0.06s).
IoU metric: bbox
Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.000
In training, the loss is dropping slowly to 1.15 but in evaluation, i do not get anything.
Please help me understand
cc @fmassa
The text was updated successfully, but these errors were encountered: