Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/push.yml
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ jobs:
runs-on: ubuntu-latest
strategy:
matrix:
model: [yolo11n,yolov8s-worldv2,yoloe11s]
model: [yolo11n, yolov8s-worldv2, yoloe11s]
steps:
- name: Checkout
uses: actions/checkout@v4
Expand Down
1 change: 0 additions & 1 deletion yolo11n/cog.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@ build:
gpu: false
python_version: "3.12"
system_packages:
- "libgl1-mesa-glx"
- "libglib2.0-0"
- "ffmpeg"
python_packages:
Expand Down
1 change: 0 additions & 1 deletion yoloe11s/cog.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@ build:
gpu: false
python_version: "3.12"
system_packages:
- "libgl1-mesa-glx"
- "libglib2.0-0"
- "ffmpeg"
python_packages:
Expand Down
9 changes: 5 additions & 4 deletions yoloe11s/predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,13 +22,14 @@ def setup(self) -> None:

def re_init_model(self, class_names: str) -> None:
"""Re-Initialize model with class names."""
if class_names.strip() != "":
if not class_names.strip():
# Load YOLOE-11s model prompt free model into memory
self.model = YOLOE("yoloe-11s-seg-pf.pt")

else:
self.model = YOLOE("yoloe-11s-seg.pt")
class_list = class_names.split(", ")
self.model.set_classes(class_list, self.model.get_text_pe(class_list))
else:
# Load YOLOE-11s model prompt free model into memory
self.model = YOLOE("yoloe-11s-seg-pf.pt")

def predict(
self,
Expand Down
1 change: 0 additions & 1 deletion yolov8s-worldv2/cog.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@ build:
gpu: false
python_version: "3.12"
system_packages:
- "libgl1-mesa-glx"
- "libglib2.0-0"
- "ffmpeg"
python_packages:
Expand Down
Loading