coding=UTF-8 from flask import Flask, render_template, request, send_from_directory from werkzeug.utils import secure_filename from iconflow.model.colorizer import ReferenceBasedColorizer from skimage.feature import canny as get_canny_feature from torchvision import transforms from PIL import Image import os import datetime import torchvision import cv2 import numpy as np import torch import einops transform_Normalize = torchvision.transforms.Compose([ transforms.Normalize(0.5, 1.0)]) ALLOWED_EXTENSIONS = set([‘png’, ‘jpg’, ‘jpeg’]) app = Flask(name) train_model = ReferenceBasedColorizer() basepath = os.path.join( os.path.dirname(file), ‘images’) # 当前文件所在路径 def allowed_file(filename): return ‘.’ in filename and filename.rsplit(‘.’, 1)[1] in ALLOWED_EXTENSIONS def load_model(log_path=‘/mnt/4T/lzq/IconFlowPaper/checkpoints/normal_model.pt’): global train_model state = torch.load(log_path) train_model.load_state_dict(state[‘net’]) @app.route(“/”, methods=[“GET”, “POST”]) def hello(): if request.method == ‘GET’: return render_template(‘upload.html’) @app.route(‘/upload’, methods=[“GET”, “POST”]) def upload_lnk(): if request.method == ‘GET’: return render_template(‘upload.html’) if request.method == ‘POST’: try: file = request.files['uploadimg'] except Exception: return None if file and allowed_file(file.filename): format = "%Y-%m-%dT%H:%M:%S" now = datetime.datetime.utcnow().strftime(format) filename = now + '_' + file.filename filename = secure_filename(filename) basepath = os.path.join( os.path.dirname(file), ‘images’) # 当前文件所在路径 # upload_path = os.path.join(basepath,secure_filename(f.filename)) file.save(os.path.join(basepath, filename)) else: filename = None return filename @app.route(‘/download/string:filename’, methods=[‘GET’]) def download(filename): if request.method == “GET”: if os.path.isfile(os.path.join(basepath, filename)): return send_from_directory(basepath, filename, as_attachment=True) pass def get_contour(img): x = np.array(img) canny = 0 for layer in np.rollaxis(x, -1): canny |= get_canny_feature(layer, 0) canny = canny.astype(np.uint8) * 255 kernel = np.array([ [0, 1, 1, 1, 0], [1, 1, 1, 1, 1], [1, 1, 1, 1, 1], [1, 1, 1, 1, 1], [0, 1, 1, 1, 0], ], dtype=np.uint8) canny = cv2.dilate(canny, kernel) # canny = Image.fromarray(canny) return canny @app.route(‘/embedding//’, methods=[“GET”, “POST”]) def icontran(img, reference): global train_model if request.method == ‘POST’: imgPath = os.path.join(basepath, img) referencePath = os.path.join(basepath, reference) img = cv2.imread(imgPath) if img is None or img.size <= 0: return None contour = get_contour(img).astype(np.float32).copy() contour = 255 - contour reference = cv2.imread(referencePath).astype(np.float32) reference = cv2.cvtColor(reference, cv2.COLOR_BGR2RGB) reference = transform_Normalize(torch.from_numpy(reference).permute(2, 0, 1).unsqueeze(0).float()/ 255.0) contour = transform_Normalize(torch.from_numpy(contour).unsqueeze(0).unsqueeze(0).float()/ 255.0) train_model.eval() transfer = train_model(contour, reference) transfer = transfer.squeeze(0) transfer = (transfer + 0.5).clamp(0, 1).mul_(255).permute(1, 2, 0).type(torch.uint8).numpy() transfer = transfer.numpy() cv2.imwrite(imgPath, transfer) return basepath # success if name == “main”: load_model() app.run(host=‘10.21.16.144’, port=9999, debug=True) 用puthon写一个调用这个服务器的gui
时间: 2023-06-05 12:07:01 浏览: 111
这是一段 Python 代码,它使用 Flask 框架来创建一个网站应用。它还导入了一些需要使用的库,如 werkzeug、numpy、torch 等。该代码可以对图片进行一些处理,如调整亮度、色彩等。其中包括一个可以将黑白图片变成彩色图片的算法 train_model。同时,代码中使用了一些图像处理的函数和变换,如 canny 算法和 einops。最后,它可以将处理后的图片保存在指定的文件夹中,并在网页中展示出来。
相关问题
#coding=utf-8 from requests_html import HTMLSession session = HTMLSession() import time import csv import os 什么意思
这段代码的作用是导入所需的模块和库,以及设置Python源文件的编码格式为UTF-8。
- `from requests_html import HTMLSession` 导入了名为`HTMLSession`的类,该类可以用于进行HTTP请求和解析HTML内容。
- `session = HTMLSession()` 创建了一个`HTMLSession`的实例对象,用于发送HTTP请求和处理响应。
- `import time` 导入了Python内置的`time`模块,用于进行时间相关的操作,如暂停程序执行。
- `import csv` 导入了Python内置的`csv`模块,用于读写CSV文件。
- `import os` 导入了Python内置的`os`模块,用于与操作系统进行交互,如文件操作和路径操作。
最后一行 `#coding=utf-8` 是一个注释,它表示该源文件的编码格式是UTF-8。这是一种常见的编码格式,用于支持多种语言和字符集。在Python 3中,通常不需要显式地声明编码格式,因为默认编码格式就是UTF-8。但是在旧版本的Python或特定环境中,可能需要添加这样的注释来确保正确的编码处理。
# coding=utf-8 和 # coding: utf-8有什么区别
两者的作用是相同的,都是用来指定Python源文件的编码格式为UTF-8。但是,它们的语法有所不同:
- `# coding=utf-8` 使用了等号,这种写法在Python 2.x中是不支持的,只有在Python 3.x中才能使用。
- `# coding: utf-8` 使用了冒号,这种写法在Python 2.x和Python 3.x中都是支持的。
在Python 2.x中,如果你想指定文件编码格式为UTF-8,应该使用第二种写法,即 `# coding: utf-8`。在Python 3.x中,两种写法都可以使用,但是推荐使用第一种写法,即 `# coding=utf-8`。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)