def encode_QR(words, picture=None, colorized=True, save_name="", save_dir=""):
if picture != None:
picture = picture.strip('''"'"''')
pair = os.path.split(picture)
if "." not in pair[1]:
print("错误的文件名！")
return
picture_full_name = pair[1].split(".")
name = picture_full_name[0]
extension = picture_full_name[1]
if save_name == "":
save_name = name + "_toQR"
if "." not in save_name and len(extension) != 0:
save_name += "." + extension
print("保存文件名为：{}".format(save_name))
save_dir = save_dir if save_dir != "" else os.getcwd()
print("保存地址为：{}".format(save_dir))
myqr.run(words,
picture=picture,
colorized=colorized,
save_name=save_name,
save_dir=save_dir)
path = PathUtil(save_dir + '\\' + save_name)
print("生成QR图片已保存到：{}".format(path.get_path()))
def load_aliyun_train_data():
aliyun_train_data = []
with open(PathUtil.get_path() + "proval/train_aliyun.txt",
"r") as file:
all_data = file.readlines()
for j in range(all_data.__len__()):
data = json.loads(all_data[j].replace('\n',
"").replace('\'', '\"'))
tags = data["response"]
tags = sorted(tags,
key=lambda x: x["confidence"],
reverse=True)
targets = data["tags"]
tag = []
for i in range(n_size):
if i < tags.__len__():
if tags[i]["tag"] in targets:
tag.append([tags[i]["confidence"], 1])
else:
tag.append([tags[i]["confidence"], 0])
else:
tag.append([0, 0])
sorted(tag, reverse=True)
aliyun_train_data.append(tag)
return aliyun_train_data
def trainRecommend():
data = json.loads(request.data.decode('utf-8'))
with open(PathUtil.get_path() + "proval/train.txt", "a+") as file:
for i in range(data.__len__()):
file.write('\n')
file.write(str(data[i]))
tag.train()
return "success"
def read_qrcode():
qr_add = input("输入QR图片地址：")
qr_add = qr_add.strip('''"'"''')
if os.path.isfile(qr_add):
path = PathUtil(qr_add)
qr_add = path.get_path()
print("正在解码。。。")
return decode_QR(qr_add)
def trainRecommend():
data = json.loads(request.data.decode('utf-8'))
with open(PathUtil.get_path() + "proval/train_aliyun.txt", "a+") as file:
with open(PathUtil.get_path() + "proval/train_baidu.txt",
"a+") as file_baidu:
for i in range(data.__len__()):
aliyun_data = {}
aliyun_data["url"] = data[i]["url"]
aliyun_data["tags"] = data[i]["tags"]
aliyun_data["response"] = data[i]["response"]
baidu_data = {}
baidu_data["url"] = data[i]["url"]
baidu_data["tags"] = data[i]["tags"]
baidu_data["response"] = data[i]["baiduResponse"]
file.write('\n')
file.write(str(aliyun_data))
file_baidu.write('\n')
file_baidu.write(str(baidu_data))
tag.train()
return "success"
def load_models(self):
self.saver.restore(self.sess,
PathUtil.get_path() + "trainmodels/model.ckpt")
def save_models(self):
self.saver.save(self.sess,
PathUtil.get_path() + "trainmodels/model.ckpt")
