search

NVIDIA-AI-IOT / tf_trt_models

TensorFlow models accelerated with NVIDIA TensorRT
BSD 3-Clause "New" or "Revised" License
686 stars 241 forks source link

Update detection.py #43

Open yuvaramsingh94 opened 5 years ago

yuvaramsingh94 commented 5 years ago

AttributeError: 'PostProcessing' object has no attribute 'score_threshold'

the pipeline.config file downloaded from your code has score_threshold inside batch_non_max_suppression block

luzhen1412 commented 5 years ago

I have changed
config.model.faster_rcnn.second_stage_post_processing.score_threshold = score_threshold
into config.model.faster_rcnn.second_stage_post_processing.batch_non_max_suppression.score_threshold = score_threshold

but the error still exists

JATINBHORE commented 4 years ago

@luzhen1412 I have encountered the same problem. did you get any solutions to that?