/ai_pytorch

Models and script for IA with PyTorch

Primary LanguagePythonMIT LicenseMIT

AI PyTorch

A library collection of PyTorch wrappers for IA training and deploy.

Dependencies

  • numpy: The fundamental package for scientific computing with Python.
  • pillow: Image processing library, the default of PyTorch;
  • pytorch: Deep Learning framework of choice;
  • onnx: Open Neural Network Exchange.

Scripts

Set Up

  1. Check out the code;
  2. Install requirements: 
    pipenv install
  3. Use as CLI, as the example below, or call the class in your project: 
    pipenv run training_siames.py -data data_path
  4. Cite me;
  5. Have fun.

To Do

  • Deployement script
  • GUI