Loading OSAP_web/OSAP_flask/core/main.py +6 −0 Original line number Diff line number Diff line from core import predict, process def s_main if __name__ == '__main__': pass No newline at end of file OSAP_web/OSAP_flask/core/predict.py +14 −0 Original line number Diff line number Diff line import os import sys import cv2 import torch #import model run in server import numpy as np os.environ["CUDA_VISIBLE_DEVICES"] = "0" torch.set_num_threads(4) device = torch.device("cuda" if torch.cuda.is_available() else "cpu") torch.cuda.empty_cache() def predict(data, model): #TODO: 模型预测，结果储存到特定文件夹中 No newline at end of file OSAP_web/OSAP_flask/core/process.py +1 −0 Original line number Diff line number Diff line #preprocess of upload image def preprocess(data_path): OSAP_web/OSAP_vue/src/assets/03_gt_dir.png 0 → 100644 +52.2 KiB Loading image diff... OSAP_web/OSAP_vue/src/assets/a.pngdeleted 100644 → 0 −44.3 KiB Loading image diff... Loading
OSAP_web/OSAP_flask/core/main.py +6 −0 Original line number Diff line number Diff line from core import predict, process def s_main if __name__ == '__main__': pass No newline at end of file
OSAP_web/OSAP_flask/core/predict.py +14 −0 Original line number Diff line number Diff line import os import sys import cv2 import torch #import model run in server import numpy as np os.environ["CUDA_VISIBLE_DEVICES"] = "0" torch.set_num_threads(4) device = torch.device("cuda" if torch.cuda.is_available() else "cpu") torch.cuda.empty_cache() def predict(data, model): #TODO: 模型预测，结果储存到特定文件夹中 No newline at end of file
OSAP_web/OSAP_flask/core/process.py +1 −0 Original line number Diff line number Diff line #preprocess of upload image def preprocess(data_path):
