update

utils.py

@@ -88,20 +88,27 @@ def infer_uploaded_image(conf, model):
                 names = res[0].names
 
                 dnames = []
+                ndetected = {}
                 for cid in dclasses:
+                    if names[cid] in ndetected:
+                        count = ndetected[names[cid]]
+                    else:
+                        count = 0
+                    ndetected.update( {names[cid]:count+1} )
                     dnames.append(names[cid])
 
-                #print('names', names)
                 res_plotted = res[0].plot()[:, :, ::-1]
 
                 with col2:
                     st.image(res_plotted,
-                             caption="Detected:{}".format(','.join(dnames)),
+                             caption="{} classes detected".format(len(dnames)),
                              use_column_width=True)
                     try:
                         with st.expander("Detection Results"):
-                            for box in boxes:
-                                st.write(box.xywh)
+                            #for box in boxes:
+                            #    st.write(box.xywh)
+                            for cname in ndetected:
+                                st.write("{} x{}".format(cname, ndetected[cname]))
                     except Exception as ex:
                         st.write("No image is uploaded yet!")
                         st.write(ex)