This repository implements various machine learning models with Python/Tensorflow. I treat mainly "Image processing" in it.
| Model | Paper | Status |
|---|---|---|
| U-Net | https://arxiv.org/abs/1505.04597 | ✅ |
| ACoL | https://arxiv.org/abs/1804.06962 | ✅ |
| Arcface | https://arxiv.org/abs/1801.07698 | ✅ |
You can use these models for training or validation.
- Python 3.6>=
- Tensorflow 2.4.0>=
- PIL
- Imgaug
- Numpy
- Scipy
- Matplotlib
I am managing these libraries with pipenv. If you do not have pipenv, install with pip
pip install pipenvYou can see latest document to understand the usage more
To install all libraries, you run
$ pipenv install
How to use each model is written in README in the each model. Basically you can training with
$ pipenv run python -m $(MODEL_NAME)/train $(options)
- Modularize this repository to enable users to import whole models
"ML models" is licenced under the MIT licence.
(C) Copyright 2023, Yudai Hayashi