Detectron.pytorch icon indicating copy to clipboard operation
Detectron.pytorch copied to clipboard

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

Results 100 Detectron.pytorch issues
Sort by recently updated
recently updated
newest added

pytorch0.4.1 cuda9+cudnn7 After saving last step checkpoint, detectron just got stucked. I manually killed it, the error infos shows it may be the deadlock in dataloader, can anyone help me?

### Expected results Will there be a Mask-X-RCNN structure for training in the future which is introduced in "Learning to Segment Everything"

Not really an issue, just want to share my experience. If you are using the code in some clusters, cuda might not be installed under `/usr/local/cuda/`. In this case, in...

compile

I want to use imgaug package to do the extra data aug ,but I do not know where to add , can anyone give me some suggestions?

Hey everyone, I am trying to get the features used for the masks. I have looked in roi_feature_transform and mask_net in model_builder.py However, I don't understand the output of the...

when I use your detectron ,I find my mask_loss didn't decrease? what happen ,and why?

Hello! I am trying to train on my own dataset, and I got a loss like below: > [Dec21-15-15-54_user-SYS-7048GR-TR_step][FPN_SE_ARP.yml][Step 201 / 150000] loss: 0.041566, lr: 0.002250 time: 1.166229, eta: 2...

Hello! I want to train a model with some samples that has no objects ( just background ). Can this implementation handle that?