Add FastAPI server

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+models/best.pt filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,174 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# UV
+#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#uv.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/latest/usage/project/#working-with-version-control
+.pdm.toml
+.pdm-python
+.pdm-build/
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+# PyPI configuration file
+.pypirc
+
+output/
+uploads/

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+RUN pip uninstall -y opencv-python && pip install --no-cache-dir opencv-python-headless
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,53 @@
+import os
+import uuid
+import yaml
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import JSONResponse
+from ultralytics import YOLO
+
+CONFIG_PATH = "./models/config.yaml"
+WEIGHT_PATH = "./models/best.pt"
+DATA_PATH = "./models/data.yaml"
+
+# Load class names from data.yaml
+with open(DATA_PATH, "r") as f:
+    data_config = yaml.safe_load(f)
+    CLASS_NAMES = data_config.get("names", [])
+
+# Load YOLO model
+model = YOLO(CONFIG_PATH, task="detect").load(WEIGHT_PATH)
+
+app = FastAPI()
+
+UPLOAD_FOLDER = "uploads"
+OUTPUT_FOLDER = "output"
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+os.makedirs(OUTPUT_FOLDER, exist_ok=True)
+
+@app.post("/predict")
+async def predict(image: UploadFile = File(...)):
+    filename = str(uuid.uuid4()) + "_" + image.filename
+    img_path = os.path.join(UPLOAD_FOLDER, filename)
+    
+    with open(img_path, "wb") as buffer:
+        buffer.write(await image.read())
+
+    results = model(img_path, conf=0.2)
+
+    detected_classes = set()
+
+    for result in results:
+        for box in result.boxes.data.tolist():
+            _, _, _, _, _, class_id = box
+            if 0 <= int(class_id) < len(CLASS_NAMES):
+                detected_classes.add(CLASS_NAMES[int(class_id)])
+
+    output_img_path = os.path.join(OUTPUT_FOLDER, filename)
+    results[0].save(filename=output_img_path)
+
+    os.remove(img_path)
+
+    if not detected_classes:
+        return JSONResponse(content={"error": "No objects detected"}, status_code=204)
+
+    return JSONResponse(content={"items": list(detected_classes), "annotated_image": output_img_path})

models/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b74959c96b6bab6eacd675a9d7e1cd618cb51f727b44d8666d745292ae6b9a47
+size 5910035

models/config.yaml

@@ -0,0 +1,50 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+# Ultralytics YOLO11 object detection model with P3/8 - P5/32 outputs
+# Model docs: https://docs.ultralytics.com/models/yolo11
+# Task docs: https://docs.ultralytics.com/tasks/detect
+
+# Parameters
+nc: 503 # number of classes
+scales: # model compound scaling constants, i.e. 'model=yolo11n.yaml' will call yolo11.yaml with scale 'n'
+  # [depth, width, max_channels]
+  n: [0.50, 0.25, 1024] # summary: 319 layers, 2624080 parameters, 2624064 gradients, 6.6 GFLOPs
+  s: [0.50, 0.50, 1024] # summary: 319 layers, 9458752 parameters, 9458736 gradients, 21.7 GFLOPs
+  m: [0.50, 1.00, 512] # summary: 409 layers, 20114688 parameters, 20114672 gradients, 68.5 GFLOPs
+  l: [1.00, 1.00, 512] # summary: 631 layers, 25372160 parameters, 25372144 gradients, 87.6 GFLOPs
+  x: [1.00, 1.50, 512] # summary: 631 layers, 56966176 parameters, 56966160 gradients, 196.0 GFLOPs
+
+# YOLO11n backbone
+backbone:
+  # [from, repeats, module, args]
+  - [-1, 1, Conv, [64, 3, 2]] # 0-P1/2
+  - [-1, 1, Conv, [128, 3, 2]] # 1-P2/4
+  - [-1, 2, C3k2, [256, False, 0.25]]
+  - [-1, 1, Conv, [256, 3, 2]] # 3-P3/8
+  - [-1, 2, C3k2, [512, False, 0.25]]
+  - [-1, 1, Conv, [512, 3, 2]] # 5-P4/16
+  - [-1, 2, C3k2, [512, True]]
+  - [-1, 1, Conv, [1024, 3, 2]] # 7-P5/32
+  - [-1, 2, C3k2, [1024, True]]
+  - [-1, 1, SPPF, [1024, 5]] # 9
+  - [-1, 2, C2PSA, [1024]] # 10
+
+# YOLO11n head
+head:
+  - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
+  - [[-1, 6], 1, Concat, [1]] # cat backbone P4
+  - [-1, 2, C3k2, [512, False]] # 13
+
+  - [-1, 1, nn.Upsample, [None, 2, "nearest"]]
+  - [[-1, 4], 1, Concat, [1]] # cat backbone P3
+  - [-1, 2, C3k2, [256, False]] # 16 (P3/8-small)
+
+  - [-1, 1, Conv, [256, 3, 2]]
+  - [[-1, 13], 1, Concat, [1]] # cat head P4
+  - [-1, 2, C3k2, [512, False]] # 19 (P4/16-medium)
+
+  - [-1, 1, Conv, [512, 3, 2]]
+  - [[-1, 10], 1, Concat, [1]] # cat head P5
+  - [-1, 2, C3k2, [1024, True]] # 22 (P5/32-large)
+
+  - [[16, 19, 22], 1, Detect, [nc]] # Detect(P3, P4, P5)

models/data.yaml

@@ -0,0 +1,508 @@
+names:
+- '-'
+- 555 sardines in tomato sauce
+- 555 tuna afritada
+- 555 tuna caldereta
+- 7up
+- absolute pure distilled drinking water
+- agnesi polenta
+- akabare khursani
+- alaska classic sweetened condensed filled milk
+- alaska condensada
+- alaska evaporada
+- all natural seasons tropical fruit
+- almond
+- almond milk
+- angel all purpose creamer
+- aook
+- apple
+- apples
+- aqua life filtration systems
+- argentina corned beef
+- argentina giniling
+- argentina meat loaf
+- artichoke
+- ash gourd
+- asparagus
+- avocado
+- bacon
+- bamboo shoots
+- banana
+- bananas
+- basmati rice
+- bazlama
+- beans
+- bear brand fortified powdered drink milk
+- bear brand swak pack chocomilk
+- beaten rice
+- beef
+- beetroot
+- bell pepper
+- bethu ko saag
+- bingo double choco
+- bingo orange cream
+- bitter gourd
+- black beans
+- black lentils
+- blackberry
+- blue lychee
+- blueberry
+- bok choy
+- book
+- bottle gourd
+- bread
+- broad beans
+- broccoli
+- brussel sprouts
+- buff meat
+- butter
+- c2 apple green tea
+- c2 lemon green tea
+- cabbage
+- cadbury dairy milk
+- cadbury dairy milk wholenut
+- calamansi
+- califlower
+- calvin klein
+- cantaloupe
+- capsicum
+- carrot
+- carrots
+- cassava
+- cauliflower
+- cdo funtastyk chicken tocino
+- cdo karne norte classic
+- cdo meat loaf
+- century tuna flakes in oil
+- cereal
+- cerealbox
+- chayote
+- cheese
+- cheez whiz original
+- chicken
+- chicken gizzards
+- chickpeas
+- chili pepper
+- chili powder
+- chilli
+- chillies
+- chips ahoy milka
+- chockolate
+- choco chum choco
+- choco chum white choco
+- choco mucho cookie sandwich white chocolate
+- choco mucho dark chocolate
+- choco mucho milk chocolate
+- choco mucho white chocolate
+- chocolate
+- chowmein noodles
+- cinnamon
+- clementine
+- cobra energy drink
+- coca cola coke zero
+- coca cola original
+- coca cola original in can
+- coca cola royal
+- coca cola sprite
+- coca cola sprite in can
+- coca cola zero sugar
+- coca cola zero sugar in can
+- coffe
+- coffee
+- combi triple choco
+- cookie
+- coriander
+- corn
+- cornflakec
+- courgates
+- crab meat
+- cream
+- cucumber
+- dabur honey
+- daily quezo
+- dates
+- datu puti soy sauce
+- datu puti vinegar
+- del monte creamy and cheesy sphagetti sauce
+- del monte fiesta fruit cocktail
+- del monte fiilipino style spaghetti sauce
+- del monte filipino style spaghetti sauce
+- del monte four seasons
+- del monte italian style spaghetti sauce
+- del monte mango juice drink
+- del monte original style tomato sauce
+- del monte pineapple juice drink heart smart
+- del monte pineapple juice with vitamins ace
+- del monte pineapple tidbits
+- del monte potato crisp biscuit original
+- del monte sweet style sphagetti sauce
+- del monte sweetened pineapple juice drink
+- delfi knick knacks strawberry
+- delight duo active probiotic fiber
+- detergent
+- don frank buko pandan gulaman powder
+- doowee donut strawberry dipped donut with strawberry flavored
+- doowee donut white choco frost with milky cream flavored
+- doreen sweetened condensed creamer
+- doritos nacho cheese
+- dr pepper
+- drinks
+- ec crunchy choco flakes
+- eden original
+- egg
+- eggplant
+- eggs
+- el rancho corned beef
+- farsi ko munta
+- ferrero rocher
+- fibisco choco mallows
+- fibisco hi ro chocolate flavored
+- fibisco the ultimate chocolate chip cookies
+- fiddlehead ferns
+- fish
+- flour
+- fudgee barr chocolate
+- fudgee barr macapuno
+- fudgee barr milk custard
+- fudgee barr vanilla
+- galia
+- galinco chips delight  chocolate chip cookies
+- galinco chips delight striped cappuccino cookies
+- galinco chips delight striped chocolate cookies
+- garden cress
+- garden peas
+- garlic
+- gatorade blue bolt
+- gatorade fierce grape
+- gatorade fierce tropical fruit
+- ginger
+- glico pocky chocolate
+- glico pocky strawberry
+- goya almond in milk chocolate
+- goya choco spread
+- goya cream white chocolate
+- goya dark chocolate
+- goya milk chocolate
+- goya take it 4 fingers
+- goya take it dark chocolate
+- goya very berries
+- goya winter chill milk chocolate
+- granny goose tortillos cheese
+- granny goose tortillos chili
+- grapes
+- great buy mushrooms pieces and stems
+- great taste supreme toffee hazelnut
+- green bell pepper
+- green brinjal
+- green chili
+- green lentils
+- green mint
+- green peas
+- green soyabean
+- groats
+- ground meat
+- gundruk
+- ham
+- hansel chocolate sandwich
+- hansel mocha sandwich
+- heavy cream
+- hokkaido mackerel
+- honey dew
+- honeydew
+- ice
+- ion supply drink pocari sweat
+- jack daniels
+- jack fruit
+- jack n jill cheesy knots milky cheese flavored
+- jack n jill chicharon ni mang juan
+- jack n jill chippy barbecue
+- jack n jill chippy chili and cheese
+- jack n jill cloud 9 choco fudge
+- jack n jill cloud 9 classic
+- jack n jill cream o vanilla
+- jack n jill dewberry blueberries n cream
+- jack n jill dewberry blueberry cheesecake
+- jack n jill hello chocolate
+- jack n jill mr chips nacho cheese flavored
+- jack n jill nips creamy tiramisu
+- jack n jill nips dark chocolate
+- jack n jill nips milk chocolate
+- jack n jill nips peanut
+- jack n jill nips white chocolate
+- jack n jill nova cheddar flavor
+- jack n jill piattos cheese
+- jack n jill piattos cheese supersized
+- jack n jill piattos roadhouse barbecue
+- jack n jill piattos roast beef
+- jack n jill piattos sour cream and onion
+- jack n jill piattos sour cream and onion supersized
+- jack n jill pic a ultimate cheddar
+- jack n jill presto choco vanilla
+- jack n jill presto chocolate
+- jack n jill presto peanut
+- jack n jill quake overload mocha butter
+- jack n jill quake overload white choco
+- jack n jill roller coaster cheddar cheese flavored
+- jack n jill vcut spicy barbeque
+- jalees bbq mushroom chicharon
+- jalees garlic mushroom chicharon
+- jersey chocolate condensed creamer
+- jersey sweetened condensed creamer
+- jinro chamisul original soju
+- jinro grapefruit soju
+- jinro green grape soju
+- jinro peach soju
+- jinro strawberry soju
+- jolly mushrooms
+- juan tasty
+- juice
+- jumpee chocolate
+- ketchup
+- kimchi
+- king sue chicken tocino
+- king sue pork tocino
+- knorr liquid seasoning original
+- kopiko brown coffee
+- ladys choice real mayonnaise
+- lapsi
+- lays classic
+- lays stax extra cheese flavour
+- lays stax original
+- lays stax sour cream and onion
+- le mineral
+- lemon
+- lemon square lava cake chocolate
+- lemon square lava cake cream bavarian
+- lemons
+- leslies clover chips cheesier
+- leslies clover chips chili and cheese
+- lettuce
+- lime
+- long beans
+- lotte pepero almond
+- lotte pepero original
+- lucky me beef na beef noodles
+- lucky me chicken na chicken noodles
+- lucky me go cup batchoy
+- lucky me go cup bulalo
+- lucky me pancit canton chilimansi
+- lucky me pancit canton extra hot chili
+- lucky me pancit canton kalamansi
+- m y san grahams
+- m-ms chocolate
+- magic chips barbeque flavored
+- magic chips cheese flavored
+- mango
+- mars caramel chocolate bar
+- masyaura
+- mayonnaise
+- mayora beng beng chocolate
+- mayora cal cheese cheese wafer
+- meat
+- milk
+- minced meat
+- mini oreo original
+- mirinda orange flavor
+- moringa leaves
+- mountain dew
+- mozarella cheese
+- mug root beer
+- mushroom
+- mussel
+- mutton
+- nagaraya garlic cracker nuts
+- nectarine
+- nescafe creamy latte twin pack
+- nescafe creamy white twin pack
+- nescafe original
+- nespresso
+- nestea lemon tea drink
+- nestle all purpose cream
+- nestle bear brand sterilized milk
+- nestle gold corn flakes
+- nestle koko krunch all in one
+- nestle koko krunch chocolate
+- nestle milo
+- nissin bread stix
+- nissin butter coconut
+- nissin cup noodles beef
+- nissin cup noodles seafood
+- nissin cup noodles spicy hot beef
+- nissin eggnog cookies
+- noodle
+- nutrela
+- nutri snack cheez it cheese
+- nutri snack moby caramel puffs
+- nutri snack moby crunchy chocolate
+- nutri snack peewee sizzling bbq
+- o puff mango cream filled marshmallows
+- oil
+- oishi bread pan buttered toast
+- oishi bread pan cheese and onion
+- oishi bread pan toasted garlic
+- oishi bread pan white cheddar cheese
+- oishi cracklings salt and vinegar
+- oishi crispy patata
+- oishi fishda fish kropeck
+- oishi kirei shrimp flavor
+- oishi pillows choco filled crackers
+- oishi pillows ube filled crackers
+- oishi potato chips plain salted flavor
+- oishi potato fries cheese
+- oishi prawn crackers spicy flavor
+- oishi rinbee cheese
+- oishi sponge chocolate crunch
+- ok cheese
+- okra
+- olive
+- olive oil
+- olives
+- onion
+- onion leaves
+- orange
+- oranges
+- oreo chocolate cream
+- oreo chocolate creme
+- oreo double stuff
+- oyster
+- paco rabanne
+- palak
+- palungo
+- pampangas best chicken longaniza
+- pampangas best native longaniza
+- pampangas best pampanga longaniza
+- pampangas best smoked longaniza
+- pampangas best tocino original
+- paneer
+- papaya
+- parmesan cheese
+- parsley
+- pasta
+- patanjali honey
+- pea
+- peach
+- pear
+- peas
+- pedro domecq fundador solera
+- pepsi
+- pik nik
+- pineapple
+- pixel 4
+- planet oat original
+- plum
+- pointed gourd
+- pork
+- pork rib
+- potato
+- pringles cheesy cheese
+- pringles pizza
+- pringles saucy bbq
+- pringles sour cream and onion
+- pumpkin
+- pure foods chinese style luncheon meat
+- queso cheese
+- radish
+- rahar ko daal
+- rasp berries
+- raspberries
+- rayo ko saag
+- rebisco choco crackers
+- rebisco hansel crackers
+- rebisco strawberry crackers
+- red beans
+- red bel pepper
+- red bell pepper
+- red lentils
+- red onion
+- refresh mineral water
+- reno liver spread
+- rice
+- ritter sport nocciole intere
+- royal elbow macaroni
+- sajjyun
+- salad
+- salmon
+- salt
+- samsung s20
+- san marino chili corned tuna
+- san marino corned tuna
+- sauce
+- sausage
+- sausages
+- scallop
+- seaweed
+- selecta choco almond fudge
+- selecta moo fortified chocolate
+- selecta quezo real
+- selecta strawberry solo pack
+- shrimp
+- sliced garlic
+- snack pack choco crunchies chocolate
+- snake gourd
+- snickers
+- soy sauce
+- soyabean
+- spinace
+- spinach
+- sponge gourd
+- spring onion
+- squash
+- stateline loaded chocolate filled
+- sticko chocolate
+- stinging nettle
+- straw berry
+- strawberry
+- sugar
+- summit drinking water
+- super delights brownie bites chocolate
+- super golden bihon
+- sweet potato
+- sweetcorn
+- swiss miss chocolate milk blend
+- tang pineapple
+- taro leaves
+- taro root
+- thukpa noodles
+- tida
+- toast bread
+- toblerone white chocolate
+- todays mixed fruit
+- tofu
+- tomato
+- tomato paste
+- topss choco
+- tori ko saag
+- tree tomato
+- tropicana
+- tuna
+- turnip
+- uni pak premium sardines
+- vicente vidal queso y cebolla
+- vitamilk choco shake
+- vitamilk double choco shake
+- vitasoy original
+- vitasoy plus original
+- wallnut
+- water
+- water melon
+- watermelon
+- wattermelon
+- wheat
+- wilkins sparkling water
+- yakult probiotics
+- yellow bell pepper
+- yellow lentils
+- yoghurt
+- yogurt
+- ys quality nata de coco green
+- yum yum strawberry snack
+- zesto big apple
+- zesto big grapes
+- zesto big mango
+- zesto big orange
+- zesto big pineapple
+- zucchini
+nc: 503
+test: ../test/images
+train: ../train/images
+val: ../valid/images

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn[standard]
+ultralytics
+python-multipart