From 5cb08121b75f30bc9e03d51cdc811cd91ae587e6 Mon Sep 17 00:00:00 2001
From: Brainway <brainway@protonmail.com>
Date: Mon, 27 Mar 2023 01:45:30 +0000
Subject: [PATCH] update ieemoo-ai-isempty.py.

---
 ieemoo-ai-isempty.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/ieemoo-ai-isempty.py b/ieemoo-ai-isempty.py
index 18bb391..0ee9ebd 100755
--- a/ieemoo-ai-isempty.py
+++ b/ieemoo-ai-isempty.py
@@ -82,18 +82,18 @@ class Predictor(object):
                 clas_ids = topN[0][0]
 
                 print("cur_img result: class id: %d, score: %0.3f" % (clas_ids, probs[0, clas_ids].item()))
-                if(int(clas_ids)==6 or int(clas_ids)==7):
-                    clas_ids = 0
-                else:
-                    clas_ids = 1
-                result["success"] = "true"
-                result["rst_cls"] = str(clas_ids)
-
-                # 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()))
+                # if(int(clas_ids)==6 or int(clas_ids)==7):
+                #     clas_ids = 0
+                # else:
+                #     clas_ids = 1
                 # result["success"] = "true"
                 # result["rst_cls"] = str(clas_ids)
 
+                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)
+
                 return result