From 82c296d18428180e1d3afbfac53b49391f0dea65 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E6=99=A8?= <lichen@ieemoo.com>
Date: Thu, 29 Dec 2022 14:00:41 +0800
Subject: [PATCH] develop update

---
 ieemoo-ai-isempty.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ieemoo-ai-isempty.py b/ieemoo-ai-isempty.py
index 49ae36c..ac9877e 100755
--- a/ieemoo-ai-isempty.py
+++ b/ieemoo-ai-isempty.py
@@ -97,7 +97,7 @@ class Predictor(object):
                 probs = torch.nn.Softmax(dim=-1)(part_logits)
                 topN = torch.argsort(probs, dim=-1, descending=True).tolist()
                 clas_ids = topN[0][0]
-                #clas_ids = 0 if 0==int(clas_ids) or 2 == int(clas_ids) or 3 == int(clas_ids) else 1
+                clas_ids = 0 if 0==int(clas_ids) or 2 == int(clas_ids) or 3 == int(clas_ids) else 1
                 #print("cur_img result: class id: %d, score: %0.3f" % (clas_ids, probs[0, clas_ids].item()))
                 result["success"] = "true"
                 result["rst_cls"] = str(clas_ids)