zhangyp15 / MonoFlex

Released code for Objects are Different: Flexible Monocular 3D Object Detection, CVPR21
MIT License
211 stars 38 forks source link

AttributeError: EDGE_FUSION_RELU #14

Open 953038395 opened 3 years ago

953038395 commented 3 years ago

when use

CUDA_VISIBLE_DEVICES=0 python tools/plain_train_net.py --config runs/monoflex.yaml --ckpt model_moderate_best_soft.pth --eval

Traceback (most recent call last): File "tools/plain_train_net.py", line 160, in args=(args,), File "/MonoFlex/engine/launch.py", line 57, in launch main_func(*args) File "tools/plain_train_net.py", line 116, in main model = KeypointDetector(cfg) File "/MonoFlex/model/detector.py", line 23, in init self.heads = bulid_head(cfg, self.backbone.out_channels) File "/MonoFlex/model/head/detector_head.py", line 29, in bulid_head return Detect_Head(cfg, in_channels) File "/MonoFlex/model/head/detector_head.py", line 13, in init self.predictor = make_predictor(cfg, in_channels) File /MonoFlex/model/head/detector_predictor.py", line 169, in make_predictor return func(cfg, in_channels) File "/MonoFlex/model/head/detector_predictor.py", line 105, in init self.edge_fusion_relu = cfg.MODEL.HEAD.EDGE_FUSION_RELU File "/.conda/envs/monoflex/lib/python3.7/site-packages/yacs/config.py", line 141, in getattr raise AttributeError(name) AttributeError: EDGE_FUSION_RELU

how to solve this problem?

zhangyp15 commented 3 years ago

Sorry, I forgot to upload the changes of the default.py. Now the problem should be fixed.