search

balancap / SSD-Tensorflow

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

Test SSD-300 model using TFRecords pipeline / ValueError: need more than 2 values to unpack (tf_ssd_bboxes_select) #45

Open busrakb opened 7 years ago

busrakb commented 7 years ago

Hi,

When SSD-Tensorflow-master/notebooks/ssd_tests.ipynb is wroked ,i am taking error that is "ValueError: need more than 2 values to unpack". when I looked at the code of ssd_common.py, i saw that tf_ssd_bboxes_select has a two output. after that i saw tf_ssd_bboxes_select_all_classes. it has tree output and pass the error.

the code nms_threshold = 0.5

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) the eror; ValueError Traceback (most recent call last)

in () 3 # Output decoding. 4 localisations = ssd.bboxes_decode(localisations, layers_anchors) ----> 5 tclasses, tscores, tbboxes = ssd_common.tf_ssd_bboxes_select(predictions, localisations) 6 tclasses, tscores, tbboxes = ssd_common.tf_bboxes_sort(tclasses, tscores, tbboxes, top_k=400) 7 tclasses, tscores, tbboxes = ssd_common.tf_bboxes_nms_batch(tclasses, tscores, tbboxes, ValueError: need more than 2 values to unpack But i took error next line, that is ssd_common.tf_bboxes_sort. the error is "AttributeError: 'module' object has no attribute 'tf_bboxes_sort' " When i looked at ssd.common.py, i couldnt find tf_bboxes_sort function. so what i need to do? can u help me? the code nms_threshold = 0.5 # Output decoding. localisations = ssd.bboxes_decode(localisations, layers_anchors) tclasses, tscores, tbboxes = ssd_common.tf_ssd_bboxes_select_all_classes(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) the error AttributeError Traceback (most recent call last) in () 4 localisations = ssd.bboxes_decode(localisations, layers_anchors) 5 tclasses, tscores, tbboxes = ssd_common.tf_ssd_bboxes_select_all_classes(predictions, localisations) ----> 6 tclasses, tscores, tbboxes = ssd_common.tf_bboxes_sort(tclasses, tscores, tbboxes, top_k=400) 7 tclasses, tscores, tbboxes = ssd_common.tf_bboxes_nms_batch(tclasses, tscores, tbboxes, 8 nms_threshold=0.3, num_classes=ssd.params.num_classes) AttributeError: 'module' object has no attribute 'tf_bboxes_sort'
ypymy commented 6 years ago

hello, have you solved this issue?

busrakb commented 6 years ago

@ypymy Sorry, I couldnt solve it :(