tien02/waste_detect_api

Provide waste detection api using Deep Learning

Waste Detectioni API

Tools

• Model: Yolov8

• Dataset: TACO trash detection dataset

• API: fastapi

Type of waste

#	Names
0	Metal
1	Plastic Bottle
2	Cigarette
3	Carton
4	Paper
5	Plastic Bag
6	Glass
7	Straw
8	Cap/Lid
9	Plastic things
10	Cup
11	Other trash

Run

Run on local

bash uvicorn_run.sh

Run on Docker

1. Build docker image

bash build.sh

2. Run docker container

bash run.sh