15 lines
553 B
Python
15 lines
553 B
Python
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)) |