Доработать нейросеть GAN pix2pix путём добавления различных механизмов

1. внедрить архитектуру нейросети STN перед основными слоями генератора U-Net архитектуры pix2pix.
Образец нейросети STN (Spatial Transformer Network) есть, неотлаженный образец интеграции есть
реализовать код, обучить на текущем датасете, получить метрики.


2. Внедрение self-attention слоя в "бутылочное горлышко" (самую узкую часть) генератора U-Net.
неотлаженный образец интеграции есть
реализовать код, обучить на текущем датасете, получить метрики.


3. перевести генератор и дискриминатор с CNN на слои внимания (модель TransGAN)
образец нейросети TransGAN на PyTorch есть, неотлаженный образец интеграции есть
реализовать код, обучить на текущем датасете, получить метрики.


Код нейросети GAN pix2pix полностью рабочий, обучаемый. Датасет есть
код написан на TensorFlow
возможно будет необходимо перевести архитектуру Нейросети GAN pix2pix с TensorFlow на PyTorch
неотлаженный образец есть


Этапная работа, оплата по сдаче каждого этапа
Предоставляется машина с видеокартой 3090Ti процессор Core i9 14900
доступ по ssh/AnyDesk


опыт работы:
PyTorch
TensoFlow
ViT
Transformers
TransGAN
pix2pix
GAN
STN (Spatial Transformer Network)
U-NET