Выставляю исходный код, так как не могу продолжать разработку нейросети. Надеюсь кому-нибудь пригодится.
Пример бы добавил. Картинка и результат
https://dtf.ru/indie/1790726-ispolzovanie-neyroseti-dlya-saundtreka-k-igram