89 lines
2.9 KiB
Python
89 lines
2.9 KiB
Python
|
import time
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
import GPUtil
|
|||
|
|
import cv2
|
|||
|
|
import numpy as np
|
|||
|
|
import torch
|
|||
|
|
from PIL import ImageFont, Image, ImageDraw
|
|||
|
|
|
|||
|
|
|
|||
|
|
# print(Path(__file__)) # 表示当前脚本文件的路径
|
|||
|
|
# print(Path(__file__).parent) # 表示当前路径的父级目录
|
|||
|
|
|
|||
|
|
|
|||
|
|
# import time
|
|||
|
|
# from contextlib import contextmanager
|
|||
|
|
#
|
|||
|
|
# @contextmanager
|
|||
|
|
# def timer():
|
|||
|
|
# start_time = time.time()
|
|||
|
|
# yield
|
|||
|
|
# end_time = time.time()
|
|||
|
|
# print('Time elapsed:', end_time - start_time)
|
|||
|
|
# # 使用上下文管理器
|
|||
|
|
# with timer():
|
|||
|
|
# time.sleep(1)
|
|||
|
|
|
|||
|
|
# print(torch.cuda.is_available())
|
|||
|
|
# print(GPUtil.getGPUs()[0].name)
|
|||
|
|
|
|||
|
|
# def get_first_gpu_name():
|
|||
|
|
# gps = GPUtil.getGPUs()
|
|||
|
|
# if gps is None or len(gps) == 0:
|
|||
|
|
# raise Exception("未获取到gpu资源, 先检测服务器是否已经配置GPU资源！")
|
|||
|
|
# return gps[0].name
|
|||
|
|
# gpu_name = get_first_gpu_name()
|
|||
|
|
# aa = [g for g in ['3090', '2080', '4090', 'A10'] if g in gpu_name]
|
|||
|
|
# print(aa)
|
|||
|
|
#
|
|||
|
|
# import tensorrt as trt
|
|||
|
|
# # 定义反序列化引擎文件的函数
|
|||
|
|
# def deserialize_engine_from_file(engine_file_path):
|
|||
|
|
# runtime = trt.Runtime(trt.Logger())
|
|||
|
|
# engine = None
|
|||
|
|
# with open(engine_file_path, "rb") as f:
|
|||
|
|
# while True:
|
|||
|
|
# data = f.read(1024 * 1024)
|
|||
|
|
# if not data:
|
|||
|
|
# break
|
|||
|
|
# tmp_engine = runtime.deserialize_cuda_engine(data)
|
|||
|
|
# if engine is None:
|
|||
|
|
# engine = tmp_engine
|
|||
|
|
# else:
|
|||
|
|
# for i in range(tmp_engine.num_bindings):
|
|||
|
|
# engine.set_binding_shape(i, tmp_engine.get_binding_shape(i))
|
|||
|
|
# return engine
|
|||
|
|
# engine_file_path = "/path/to/engine_file.engine"
|
|||
|
|
# s = time.time()
|
|||
|
|
# engine = deserialize_engine_from_file(engine_file_path)
|
|||
|
|
# print("1 加载trt文件时间", time.time() - s)
|
|||
|
|
# s1 = time.time()
|
|||
|
|
# with open(engine_file_path, "rb") as f1, trt.Runtime(trt.Logger(trt.Logger.ERROR)) as runtime:
|
|||
|
|
# model = runtime.deserialize_cuda_engine(f1.read())
|
|||
|
|
# print("2 加载trt文件时间", time.time() - s1)
|
|||
|
|
|
|||
|
|
def get_label_array(color=None, label=None, outfontsize=None, fontpath="conf/platech.ttf"):
|
|||
|
|
# Plots one bounding box on image 'im' using PIL
|
|||
|
|
fontsize = outfontsize
|
|||
|
|
font = ImageFont.truetype(fontpath, fontsize, encoding='utf-8')
|
|||
|
|
x,y,txt_width, txt_height = font.getbbox(label)
|
|||
|
|
print(x,y,txt_width, txt_height)
|
|||
|
|
im = np.zeros((txt_height, txt_width, 3), dtype=np.uint8)
|
|||
|
|
im = Image.fromarray(im)
|
|||
|
|
draw = ImageDraw.Draw(im)
|
|||
|
|
draw.rectangle([0, 0, txt_width, txt_height], fill=tuple(color))
|
|||
|
|
draw.text((0, -3), label, fill=(255, 255, 255), font=font)
|
|||
|
|
im_array = np.asarray(im)
|
|||
|
|
# if outfontsize:
|
|||
|
|
# scaley = outfontsize / txt_height
|
|||
|
|
# im_array = cv2.resize(im_array, (0, 0), fx=scaley, fy=scaley)
|
|||
|
|
return im_array
|
|||
|
|
|
|||
|
|
|
|||
|
|
aaa = time.time()
|
|||
|
|
im_array = get_label_array(color=(0, 255, 0), label="排口", outfontsize=40, fontpath="platech.ttf")
|
|||
|
|
print(time.time() - aaa)
|
|||
|
|
cv2.imshow("frame", im_array)
|
|||
|
|
cv2.waitKey(0)
|