@apiotrowski312/image-loader

Load an image (or batch of images) and put them in Canvas for you. Pretty useful for machine learning with NodeJS and TensorFlow. Should work fine with any node thanks to @babel/polyfil.

Installation

This is a Node.js module available through the npm registry. It can be installed using the npm or yarn command line tools.

npm install @apiotrowski312/image-loader --save

Usage

This module assumes that, you have all images in one directory. All of the images with the same naming convention (for creating labels).

Basic use

import LoadImages from '../src/index';

const imageLoader = new LoadImages('./images', { one: 'first', two: 'second'}, {silent: true, batchSize: 10});
const loadOneImage = imageLoader.loadLabelAndImage();
const loadBatchOfImages = imageLoader.loadBatchSizeImages();

Detailed options

new LoadImages(PATH, LABELS, OPTIONS);

• PATH - path as a string
• LABELS - Object, with key - value pairs. Each label should match your labelRegExp option.

For example, if you have a directory with images named like (and you didn't change labelRegExp):

one.1.jpg, one.2.jpg, two.jpg, two.xxx.qqq.png,

then your labels key should look simmilar to following example. Values can be different, following example would work well with TensorFlowJS.

const labels = {
  one: [1, 0],
  two: [0, 1],
}

• OPTIONS

KEY	DEFAULT	DESCRIPTION
shuffle	true	shuffle order of loading images if true
shufflePhrase	'phrase'	shuffle phrase for shuffle-seed
batchSize	10	how much images do you want load in one batch
trainSize	100	how much images do you want load to the script
labelRegExp	RegExp(/[^.]*/)	regexp for matching labels from img filenames
extentionsRegExp	`RegExp(/.(jpg	png
silentMode	true	if true then there will be no console.logs(). Usesfull for debugging.

Tests

npm install
npm test

Dependencies

• @babel/polyfill: Provides polyfills necessary for a full ES2015+ environment
• canvas: Canvas graphics API backed by Cairo
• package-json-to-readme: Generate a README.md from package.json contents
• shuffle-seed: Shuffle Array based on a Seed

Dev Dependencies

• @babel/cli: Babel command line.
• @babel/core: Babel compiler core.
• @babel/node: Babel command line
• @babel/preset-env: A Babel preset for each environment.
• babel-core: A placeholder package that bridges babel-core to @babel/core.
• eslint: An AST-based pattern checker for JavaScript.
• eslint-config-airbnb: Airbnb's ESLint config, following our styleguide
• eslint-config-prettier: Turns off all rules that are unnecessary or might conflict with Prettier.
• eslint-plugin-import: Import with sanity.
• eslint-plugin-jsx-a11y: Static AST checker for accessibility rules on JSX elements.
• eslint-plugin-prettier: Runs prettier as an eslint rule
• eslint-plugin-react: React specific linting rules for ESLint
• prettier: Prettier is an opinionated code formatter
• jest: Delightful JavaScript Testing.

To Do

• [ ] add JSDoc (or simmilar)
• [ ] remove trainSize option
• [ ] change throwing Errors to something else (maybe return undefined?)

License

MIT