lichen/ieemoo-ai-isempty
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update ieemoo-ai-isempty.py.

Browse Source
This commit is contained in:
Brainway
2022-10-11 07:38:14 +00:00
committed by Gitee
parent d5504fce91
commit 062e2245de
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ieemoo-ai-isempty.py
View File

@ -54,7 +54,7 @@ def parse_args(model_file="../module/ieemoo-ai-isempty/model/now/emptyjudge5_che
class Predictor(object):
def __init__(self, args):
self.args = args
self.args.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
self.args.device = torch.device("cpu")
#print(self.args.device)
self.args.nprocs = torch.cuda.device_count()
self.cls_dict = {}
@ -121,7 +121,7 @@ def get_isempty():
#print('get date use time: {0:.2f}s'.format(getdateend - start))
pic = json_data.get("pic")
result = { }
result = {}
imgdata = base64.b64decode(pic)
imgdata_np = np.frombuffer(imgdata, dtype='uint8')