s456501/Plankton_Detector
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change model weights location

Browse Source
This commit is contained in:
Piotr Szkudlarek 2024-01-08 14:05:02 +01:00
parent bb037a4f68
commit 42469f3572
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
PlanktonDetector/DetectionApp/utils.py
View File

@ -5,7 +5,7 @@ from io import BytesIO
from django.conf import settings from django.conf import settings
from ultralytics import YOLO from ultralytics import YOLO
MODEL = YOLO("best.pt") MODEL = YOLO("../best.pt")
def predict_image(image): def predict_image(image):