An open API service indexing awesome lists of open source software.

https://github.com/SunchhayK/Product_Image_Classification

This project aims to take an input of a product image and then classify it to a certain category by using a supervised learning approach.
https://github.com/SunchhayK/Product_Image_Classification

classification machine-learning pytorch tensorflow

Last synced: 5 months ago
JSON representation

This project aims to take an input of a product image and then classify it to a certain category by using a supervised learning approach.

Awesome Lists containing this project

README

          

# Product Image Classification

This project aims to take an input of a product image and then classify it to a certain category by using a supervised learning approach.

## Requirements

| Name | Version |
|-|-|
| *Python* | **3.10.*** |

## Run the project

Create a conda virtual environment.

### Install the dependencies

```py
pip install -r requirements.txt
```

### Start the project

```py
jupyter notebook
```

> The `main.ipynb` is the main file of our project. It outlines how we pre-process our dataset and all the different types of models (the best version) we have trained is evaluated and compared in this file.

## Author

- Khoun Sovansunchhay
- John Liza
- Chan Liza