I want to find out how class mapping works for a fine tuning task.
Denote that we have the custom dataset with class such as “motorcycle”.
We also have weight of TrafficCamNet, that was trained for class of “two-wheelers”.
How should I write target_class_mapping
to the specs file to train the model on a custom dataset using knowledge of the pre-trained TrafficCamNet?
What option will lead me to the goal?
target_class_mapping {
key: "motorcycle"
value: "two-wheelers"
}
OR
target_class_mapping {
key: "two-wheelers"
value: "motorcycle"
}
OR
target_class_mapping {
key: "motorcycle"
value: "motorcycle"
}