Can this implementation handle images without objects?

roytseng-tw / Detectron.pytorch

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

MIT License

2.82k stars 565 forks source link

Open SingL3 opened 5 years ago

SingL3 commented 5 years ago

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

13070151771 commented 5 years ago

可以

SingL3 commented 5 years ago

@13070151771 意思是只要把 roidb = filter_for_training(roidb) 去掉就可以吗？我看代码，大多数都对无目标图可以处理，但是有少部分似乎就无法处理，改完之后loss经常突然增大