zzzxxxttt / simple_detectron2_efficientdet Goto Github PK

Hello,

Could I ask you the FPS you obtain with detectron_efficientNet on COCO dataset?
How I can set the backbone EfficientNet b1,...,b3

Thank you

How do I change number of classes here? And say, I have 17 classes, what should be the number?

@zzzxxxttt
how to predict a single image

Hi,
I got this following error when I run the code:

-- Process 0 terminated with the following error:
Traceback (most recent call last):
File "/opt/conda/lib/python3.7/site-packages/torch/multiprocessing/spawn.py", line 20, in _wrap
fn(i, *args)
File "/root/jwei/detectron2/detectron2/engine/launch.py", line 94, in _distributed_worker
main_func(*args)
File "/root/jwei/detectron2/projects/Onlyou/train_net.py", line 94, in main
trainer = Trainer(cfg)
File "/root/jwei/detectron2/detectron2/engine/defaults.py", line 282, in init
model = self.build_model(cfg)
File "/root/jwei/detectron2/detectron2/engine/defaults.py", line 440, in build_model
model = build_model(cfg)
File "/root/jwei/detectron2/detectron2/modeling/meta_arch/build.py", line 21, in build_model
model = META_ARCH_REGISTRY.get(meta_arch)(cfg)
File "/root/jwei/detectron2/projects/Onlyou/model/efficientnet/retinanet.py", line 27, in init
self.device = torch.device(cfg.MODEL.DEVICE)
File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 638, in setattr
object.setattr(self, name, value)
AttributeError: can't set attribute

Could you pls give me advice ? thanks