/GANTheftAutoWithAOgmaNeo

Based on Sentdex's "GAN Theft Auto", but using AOgmaNeo instead of a GAN

Primary LanguagePythonMIT LicenseMIT

GANTheftAutoWithAOgmaNeo

These are the training scripts for training an AOgmaNeo model on Sentdex's "GAN Theft Auto" dataset. They were used to produce this video.

Sentdex's original repository, which includes the sample dataset, is here.

These scripts assume PyAOgmaNeo is installed, as well as OpenCV.

To use, make sure the dataset is in a folder called "gtagan_2_sample" within this folder, and then run neoTrainer.py. It will periodically save out the training progress when it says "Saved", after which you can run neoPlayer.py to see the result so far.

MIT License