According to the training spec file, the custom model is trained on two classes (car and two-wheelers). So it can only detect the cars and two-wheelers. In order to detect four classes, need to prepare corresponding dataset and set four classes in training spec for training.
You aims to train a two classes model, right?
Your training dataset(9227 images) contains 29684 cars and only 1818 two-wheelers.
Could you train via
- set the lr rate very small
(min_learning_rate: 5e-06
max_learning_rate: 5e-05 ) - disable enable_autoweighting
- set higher class weight (0.2) for two-wheelers class and lower class weight (0.01) for car class