|
- import multiprocessing as mp
- import time
- from concurrent.futures import ProcessPoolExecutor, ThreadPoolExecutor
- from multiprocessing import Queue, shared_memory
-
- import tensorrt as trt
-
- # multiprocessing.set_start_method('spawn')
- Detweights = "/home/th/tuo_heng/dev/AIlib2/weights/river2/yolov5_2080Ti_fp16.engine"
- with open(Detweights, "rb") as f:
- model = f.read()
- Segweights = "/home/th/tuo_heng/dev/AIlib2/weights/river2/stdc_360X640_2080Ti_fp16.engine"
- with open(Segweights, "rb") as f:
- segmodel = f.read()
- print(type(model), type(segmodel))
|