我在python2中开发了一些应用程序,使用caffe作为框架来对某些类型的图像进行分类。这种系统的输出应该是一个numpy数组,它的值是softmax层的概率,我应该把最大值和它的索引作为预测。
但是,有时(如200次中的3次)概率值不会以十进制表示形式打印。相反,它是以字典的形式。
例如,我得到了0.02345
而不是{s: 1, e: -3, c: [2, 3, 4, 5]}
for i, _ in enumerate(images):
output_prob = self.net.blobs['prob'].data[i]
top_indexes = output_prob.argsort()[::-1][:num_of_predictions]
predictions = [(self.labels[j], float(output_prob[j]))
for j in top_indexes]
results.append(predictions)
有人经历过同样的事吗?有关其他信息,服务器使用带有gunicorn作为Web服务器的烧瓶实现。
谢谢!
=============================================== ===========================
编辑:这是负责值
的代码片段results = {}
for i, pred in enumerate(predictions_all):
predictions = [{'class': x[0], 'confidence': x[1]} for x in pred]
results[image_urls[i]] = dict(predictions=predictions)
return jsonify(results), 200
,这在不同的文件中
<ListView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:dividerHeight="10dp"
android:id="@+id/list_primary">
</ListView>
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:id="@+id/text14"
android:text="Taken"
android:textColor="@color/black"
android:layout_marginLeft="20dp"
android:layout_marginTop="10dp"
android:textSize="20sp"/>
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:id="@+id/text16"
android:text="Skip"
android:textColor="@color/black"
android:layout_marginTop="-25dp"
android:layout_gravity="center"
android:textSize="20sp"/>
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="Dismiss"
android:id="@+id/text15"
android:textColor="@color/black"
android:layout_gravity="right"
android:layout_marginTop="-25dp"
android:layout_marginRight="20dp"
android:textSize="20sp"
/>
</LinearLayout>
<LinearLayout
android:orientation="vertical"
android:layout_width="fill_parent"
android:layout_height="match_parent">
<Button
android:layout_width="70dp"
android:layout_height="70dp"
android:background="@mipmap/btn1"
android:layout_marginLeft="15dp"
android:id="@+id/btn_taken" />
<Button
android:layout_width="70dp"
android:layout_height="70dp"
android:background="@mipmap/btn2"
android:id="@+id/btnskip"
android:layout_gravity="center"
android:layout_marginTop="-70dp"
/>
<Button
android:layout_width="70dp"
android:layout_height="70dp"
android:background="@mipmap/btn3"
android:id="@+id/btndismiss"
android:layout_marginRight="20dp"
android:layout_gravity="right"
android:layout_marginTop="-70dp"
/>
</LinearLayout>
相关库的完整版本是numpy == 1.11.1 flask == 0.11.1 gunicorn == 19.6.0