Remove car and bicycle classes from Xavier sample config files

Hello,

How can the car and bicycle classes can be removed from the Xavier config file? We would like to test only the person class.
Thank you.

Hi,

It’s required to refine the model with the corresponding database for your usecase.
Here is our training tutorial for your reference: https://github.com/NVIDIA/DIGITS

If retraining is not an option for you, an alternative is to remove the labels.txt as a workaround.
Edit /home/nvidia/Model/ResNet_18/labels.txt

diff --git a/labels.txt b/labels.txt
index 656a052..7bfe817 100644
--- a/labels.txt
+++ b/labels.txt
@@ -1,3 +1 @@
 Person
-RS
-Vehicle

Thanks.

Can you be more specific on the directory because we can’t find it. Thank you

Hi,

As mentioned, please find the file at ‘/home/nvidia/Model/ResNet_18/labels.txt’.

Thanks.