How to hide certain category while inferencing

[gyumin0514]

Hi. I trained my object detection model and it works well. I got 5 classes, and now I want to hide 3 classes including bbox when inferencing. For example, I trained person, dog, cat, car and chair, and even if the test image I want to inference, I only want model to visualize person and chair. Which part of which file should I fix?