about the infer.py AttributeError: BBOX_AUG

[gsscumtseu]

Thank you for sharing such nice work.
Although while doing inference I noticed one anomaly (but maybe I am missing some settings). There is error happened:

Traceback (most recent call last):
File "/home/gss/CarLocFiles/Occlusion_Net_Test/infer.py", line 152, in
main()
File "/home/gss/CarLocFiles/Occlusion_Net_Test/infer.py", line 80, in main
coco_demo = COCODemo(
File "/home/gss/CarLocFiles/Occlusion_Net_Test/lib/predictor.py", line 111, in init
self.model = build_detection_model(cfg)
File "/home/gss/CarLocFiles/Occlusion_Net_Test/lib/occlusion_net/detector/detectors.py", line 10, in build_detection_model
return meta_arch(cfg)
File "/home/gss/CarLocFiles/Occlusion_Net_Test/lib/occlusion_net/detector/generalized_rcnn.py", line 31, in init
self.roi_heads = build_roi_heads(cfg, self.backbone.out_channels)
File "/home/gss/CarLocFiles/Occlusion_Net_Test/lib/occlusion_net/roi_heads/roi_heads.py", line 90, in build_roi_heads
roi_heads.append(("box", build_roi_box_head(cfg, in_channels)))
File "/home/gss/CarLocFiles/Occlusion_Net_Test/maskrcnn_benchmark_1/maskrcnn_benchmark/modeling/roi_heads/box_head/box_head.py", line 71, in build_roi_box_head
return ROIBoxHead(cfg, in_channels)
File "/home/gss/CarLocFiles/Occlusion_Net_Test/maskrcnn_benchmark_1/maskrcnn_benchmark/modeling/roi_heads/box_head/box_head.py", line 21, in init
self.post_processor = make_roi_box_post_processor(cfg)
File "/home/gss/CarLocFiles/Occlusion_Net_Test/maskrcnn_benchmark_1/maskrcnn_benchmark/modeling/roi_heads/box_head/inference.py", line 162, in make_roi_box_post_processor
bbox_aug_enabled = cfg.TEST.BBOX_AUG.ENABLED
File "/home/gss/anaconda3/envs/Occlusion_Net/lib/python3.8/site-packages/yacs/config.py", line 141, in getattr
raise AttributeError(name)
AttributeError: BBOX_AUG