AlexNet-Swift icon indicating copy to clipboard operation
AlexNet-Swift copied to clipboard

An implementation of the standard AlexNet convolutional network in Swift for TensorFlow

AlexNet in Swift

This is an implementation of the classic AlexNet convolutional neural network architecture in Swift for TensorFlow.

It currently builds and runs on both the Mac (via an Xcode project) and on Ubuntu 18.04 (via the swift build command in the main directory). The application has been updated to be current with the Swift for TensorFlow API as of the April 18, 2019 snapshot.

On the Mac, you may need to open up your Xcode's File | Project Settings... menu option and switch the Build System to Legacy Build System to get this to compile.

On Ubuntu, running swift build will deposit the binary in the .build directory, so to run this against the weights and images that ship with the project, you'll want to run the following from the base project directory:

.build/x86_64-unknown-linux/debug/AlexNet