You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: examples/kenning-instance-segmentation/README.md
+17-4Lines changed: 17 additions & 4 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -85,7 +85,6 @@ Also, if you want to change the camera path, set the `CAMERA_PATH` variable with
85
85
Scripts checks for the presence of NVIDIA drivers - if NVIDIA GPU is not present, the container will run in CPU-only mode.
86
86
If you want to explicitly run the container without GPU acceleration, run:
87
87
88
-
89
88
```
90
89
../run-docker.sh cpu
91
90
```
@@ -124,10 +123,24 @@ cd /data
124
123
To get the runtime to perform in real time, you need to first optimize YOLACT model for your machine.
125
124
In this example, we will use the Apache TVM compiler to compile the model for the GPU.
126
125
126
+
### Install TVM for CPU only inference
127
+
128
+
If you run CPU only inference you need to install:
129
+
130
+
```bash
131
+
pip install --force-reinstall apache-tvm
132
+
```
133
+
134
+
and then run:
135
+
136
+
```bash
137
+
ldconfig
138
+
```
139
+
127
140
First, install Kenning with its necessary dependencies:
128
141
129
142
```bash
130
-
pip install ./kenning[object_detection]
143
+
pip install "./kenning[object_detection]"
131
144
```
132
145
133
146
Then, run the [scenario with YOLACT optimizations for the GPU](https://github.com/antmicro/kenning/blob/main/scripts/jsonconfigs/yolact-tvm-gpu-detection.json).
@@ -168,7 +181,7 @@ Then, to run the demo, load the ROS 2 environment including the newly built pack
168
181
source install/setup.sh
169
182
```
170
183
171
-
Next, launch Kenning, Camera node, and GUI node using the launch file [`kenning-instance-segmentation.py`](./kenning-instance-segmentation.py):
184
+
Next, launch Kenning, Camera node, and GUI node using the launch file [`kenning-instance-segmentation.py use_gui:=True`](./kenning-instance-segmentation.py):
For CPU-only demo of the instance segmentation run [`kenning-instance-segmentation-cpu.py`](./kenning-instance-segmentation-cpu.py) as follows:
196
+
For CPU-only demo of the instance segmentation run [`kenning-instance-segmentation-cpu.py use_gui:=True`](./kenning-instance-segmentation-cpu.py) as follows:
0 commit comments