Upload nodes.py
Browse files
nodes.py
CHANGED
@@ -365,7 +365,8 @@ from ultralytics import YOLO
|
|
365 |
from PIL import Image
|
366 |
|
367 |
# Load a pretrained YOLOv8n model
|
368 |
-
|
|
|
369 |
|
370 |
class Mynode_2:
|
371 |
@classmethod
|
|
|
365 |
from PIL import Image
|
366 |
|
367 |
# Load a pretrained YOLOv8n model
|
368 |
+
current_directory = os.getcwd()
|
369 |
+
model = YOLO(task='detect', model=current_directory + '/custom_nodes/yolov8_face/yolov8m_200e.pt')
|
370 |
|
371 |
class Mynode_2:
|
372 |
@classmethod
|