GAN
Установка пакетов:
pip requirements.txt
Препроцессинг данных:
python preporocessing.py
--dataset_name=spine_segmentation
--dataset_dir=./datasets/spine_segmentation/spine_segmentation_1/test/
--output_name=spine_segmentation_train_1_fold
--output_dir=./datasets/tfrecords_spine_segmentation_with_superpixels
Тренировка:
python train_gan.py
Можно также задать директорию для выода результатов --train_dir, директорию с датасетом--dataset_dir, --batch_size, --optimizer, --num_epochs