36 lines
831 B
Python
36 lines
831 B
Python
import time
|
|
from sophon_chakcy.utils.task_manager import TaskManager, TaskArgs
|
|
from module_bank import PythonToSQLite
|
|
|
|
# 安装导入器
|
|
packer = PythonToSQLite("plugins.mbank")
|
|
finder = packer.install_importer()
|
|
|
|
|
|
def main():
|
|
task_manager = TaskManager(4)
|
|
start_time = time.time()
|
|
print(start_time)
|
|
for i in range(1):
|
|
task_name = f"yolo_test_{i}"
|
|
args = TaskArgs(
|
|
task_name=task_name,
|
|
algorithms=["yolo_example", "yolo_example"],
|
|
input="rtsp://192.168.137.1:8554/stream",
|
|
is_vfs=True
|
|
)
|
|
task_manager.submit(args)
|
|
|
|
# time.sleep(5)
|
|
|
|
# for i in range(4):
|
|
# task_name = f"yolo_test_{i}"
|
|
# task_manager.stop_task(task_name)
|
|
|
|
# print(time.time() - start_time)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|