Swiftpack.co - wvabrinskas/NeuronDatasets as Swift Package

Swiftpack.co is a collection of thousands of indexed Swift packages. Search packages.
See all packages published by wvabrinskas.
wvabrinskas/NeuronDatasets 1.0.13
Machine learning datasets to be used with Neuron package
⭐️ 1
🕓 5 weeks ago
iOS macOS watchOS tvOS
.package(url: "https://github.com/wvabrinskas/NeuronDatasets.git", from: "1.0.13")


Package contains machine learning datasets that are supported the Neuron package.

Supported Datasets

Dataset Origin
CIFAR-10 Local
QuickDraw Remote

Importing image datasets

You can import images from a directory to create a Dataset that Neuron can use. Useful for datasets you can download from Kaggle. Mostly useful for GAN and other generative networks.

Create an ImageDataset object

  let dataset = ImageDataset(imagesDirectory: URL(string: "/Users/williamvabrinskas/Desktop/ImageDataset")!,
                             imageSize: CGSize(width: 64, height: 64),
                             label: [1.0],
                             imageDepth: .rgb,
                             maxCount: 10000)
  • imagesDirectory: The directory of the images to load. All images should be the same size.
  • imageSize: The expected size of the images
  • label: The label to apply to every image.
  • imageDepth: ImageDepth that describes the expected depth of the images.
  • maxCount: Max count to add to the dataset. Could be useful to save memory. Setting it to 0 will use the whole dataset.
  • validationSplitPercent: Number between 0 and 1. The lower the number the more likely it is the image will be added to the training dataset otherwise it'll be added to the validation dataset.
  • zeroCentered: Format image RGB values between -1 and 1. Otherwise it'll be normalized to between 0 and 1.

To build the dataset just call .build() on the dataset object.


In the bin folder there are some helpful scripts to help format image databases.

Script Description Usage
resize.py will automatically resize images in a given directory to a specified size python3 ./bin/resize.py --width 64 --height 64 --path PATH_TO_IMAGES_DIR


Stars: 1
Last commit: 5 weeks ago
jonrohan Something's broken? Yell at me @ptrpavlik. Praise and feedback (and money) is also welcome.


Release Notes

5 weeks ago

Updated Neuron version

Swiftpack is being maintained by Petr Pavlik | @ptrpavlik | @swiftpackco | API | Analytics