ABCnetv2测试报错

[tkone2018]

你好，我在测试abcnetv2的时候报了这个错 Traceback (most recent call last): File "demo/demo.py", line 75, in demo = VisualizationDemo(cfg) File "/root/code/adet/demo/predictor.py", line 40, in init self.predictor = DefaultPredictor(cfg) File "/root/code/detectron2/detectron2/engine/defaults.py", line 182, in init self.model = build_model(self.cfg) File "/root/code/detectron2/detectron2/modeling/meta_arch/build.py", line 21, in build_model model = META_ARCH_REGISTRY.get(meta_arch)(cfg) File "/root/code/adet/adet/modeling/one_stage_detector.py", line 72, in init super().init(cfg) File "/root/code/detectron2/detectron2/config/config.py", line 147, in wrapped explicit_args = _get_args_from_config(from_config_func, *args, *kwargs) File "/root/code/detectron2/detectron2/config/config.py", line 180, in _get_args_from_config ret = from_config_func(args, **kwargs) File "/root/code/detectron2/detectron2/modeling/meta_arch/rcnn.py", line 75, in from_config backbone = build_backbone(cfg) File "/root/code/detectron2/detectron2/modeling/backbone/build.py", line 31, in build_backbone backbone = BACKBONE_REGISTRY.get(backbone_name)(cfg, input_shape) File "/root/code/adet/adet/modeling/backbone/bifpn.py", line 389, in build_fcos_resnet_bifpn_backbone backbone = BiFPN( File "/root/code/adet/adet/modeling/backbone/bifpn.py", line 331, in init self.repeated_bifpn.append(SingleBiFPN( File "/root/code/adet/adet/modeling/backbone/bifpn.py", line 146, in init raise NotImplementedError NotImplementedError 帮忙分析下，谢谢

[Yuliang-Liu]

@tkone2018 Did you solved this issue?

[hungnt14]

I also faced this issue and solved it by rechecking my train data.

In my case, the problem is caused by an image with a weird size (4000px x 160px). Then I removed all images that have an aspect ratio larger than 10 and the train was smooth again.

You can try to recheck your training data if you got this error.

[tkone2018]

@hungnt14 ok，thx, i will try it