[#53928] cvnode_base: nodes: requirements.txt: Add python packages re…

…quirements Signed-off-by: Illia Vysochyn <[email protected]>
antmicro · Jan 16, 2024 · b5c88c4 · b5c88c4
1 parent 227ae30
commit b5c88c4
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/cvnode_base/nodes/requirements.txt b/cvnode_base/nodes/requirements.txt
@@ -0,0 +1,12 @@
+cuda-python
+detectron2 @ git+https://github.com/facebookresearch/detectron2.git@8c4a333ceb8df05348759443d0206302485890e0
+kenning[object_detection,reports] @ git+https://github.com/antmicro/kenning.git
+nvidia-pyindex
+onnx
+onnx_graphsurgeon
+onnxruntime
+opencv-python
+tensorrt
+torch
+torchvision
+typing_extensions==4.4.0