search

balancap / SSD-Tensorflow

Single Shot MultiBox Detector in TensorFlow
4.11k stars 1.89k forks source link

In jupyter notebook , AttributeError #242

Open DRosemei opened 6 years ago

DRosemei commented 6 years ago

AttributeError: module 'nets.ssd_common' has no attribute 'tf_bboxes_sort' in ssd_test input[18]

Output decoding.

localisations = ssd.bboxes_decode(localisations, layers_anchors) tclasses, tscores, tbboxes = ssd_common.tf_ssd_bboxes_select(predictions, localisations) tclasses, tscores, tbboxes = ssd_common.tf_bboxes_sort(tclasses, tscores, tbboxes, top_k=400) tclasses, tscores, tbboxes = ssd_common.tf_bboxes_nms_batch(tclasses, tscores, tbboxes, nms_threshold=0.3, num_classes=ssd.params.num_classes)

I can't find tf_bboxes_sort and tf_bboxes_nms_batch in ssd_common.py. Besides, ssd_common.tf_ssd_bboxes_select only return two args. Could someone tell me why? @balancap

houkai commented 6 years ago

maybe you can use detected_bboxes in ssd_vgg_512.py file. As the way in eval_ssd_network.py.

those similar method can see 'nets/npmethods', but np* isn't realized in graph theory.