ML Models

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.

Requirements

• 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 pipenv

You can see latest document to understand the usage more

To install all libraries, you run

$ pipenv install

Usage

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)

Future Plans

• Modularize this repository to enable users to import whole models

Licence

"ML models" is licenced under the MIT licence.

(C) Copyright 2023, Yudai Hayashi