File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -393,9 +393,9 @@ void DetectionOutputLayer<Dtype>::Forward_cpu(
393393 }
394394 string image_name = pt.get <string>(" image_id" );
395395 float score = pt.get <float >(" score" );
396- vector<int > bbox;
396+ vector<float > bbox;
397397 BOOST_FOREACH (ptree::value_type &elem, pt.get_child (" bbox" )) {
398- bbox.push_back (static_cast <int >(elem.second .get_value <float >()));
398+ bbox.push_back (static_cast <float >(elem.second .get_value <float >()));
399399 }
400400 *(outfiles[label_name]) << image_name;
401401 *(outfiles[label_name]) << " " << score;
Original file line number Diff line number Diff line change @@ -225,9 +225,9 @@ void DetectionOutputLayer<Dtype>::Forward_gpu(
225225 }
226226 string image_name = pt.get <string>(" image_id" );
227227 float score = pt.get <float >(" score" );
228- vector<int > bbox;
228+ vector<float > bbox;
229229 BOOST_FOREACH (ptree::value_type &elem, pt.get_child (" bbox" )) {
230- bbox.push_back (static_cast <int >(elem.second .get_value <float >()));
230+ bbox.push_back (static_cast <float >(elem.second .get_value <float >()));
231231 }
232232 *(outfiles[label_name]) << image_name;
233233 *(outfiles[label_name]) << " " << score;
You can’t perform that action at this time.
0 commit comments