Вы выполняете трансферное обучение для классификации марки и модели транспортного средства, используя предварительно обученную модель, натренированную на общих изображениях. У вас есть большой размеченный набор данных изображений транспортных средств. Как следует инициализировать модель перед переобучением на вашем наборе данных транспортных средств?

При трансферном обучении для классификации изображений наиболее эффективный подход — использовать веса предварительно обученной модели для большинства слоев. Эти веса уже научились извлекать общие признаки из изображений. Однако последний полносвязный слой (или классификационная головка) должен быть заменен новым слоем, инициализированным случайными весами, поскольку он отвечает за классификацию конкретных классов в вашем наборе данных (марок и моделей транспортных средств), которые отличаются от классов, на которых обучалась исходная модель. Инициализация всех слоев случайными весами равносильна обучению модели с нуля, что игнорирует преимущества предварительного обучения. Использование предварительно обученных весов для последнего слоя не позволит модели адаптироваться к вашим новым классам.