Предсказание Darkflow только на одном классе
Я использую Darkflow для обнаружения автомобилей на фотографиях (в конце концов, в прямом эфире видео) для школьного проекта. Йоло-сеть обучается для нескольких классов (например, велосипед, собака, ноутбук, автомобиль, автобус, грузовик). Мне нужно обнаружить только легковые и грузовые автомобили. Может быть, это может улучшить мою производительность, так как он анализирует каждую картинку в 40 или более. Подскажите, пожалуйста, как убрать ненужные классы?