diff --git a/01.加载模型.py b/01.加载模型.py index 71367d8..784c6fe 100644 --- a/01.加载模型.py +++ b/01.加载模型.py @@ -1,6 +1,8 @@ from ultralytics import YOLO -yolo = YOLO('models/yolov8n.pt') -# yolo('./resources/01.png',show=True,save=True) -# yolo('./resources/FHC09594.jpg',show=True,save=True) -yolo('./resources/02.jpg',show=True,save=True) \ No newline at end of file +# model = YOLO('models/yolov8n.pt') +model = YOLO('models/yolov8x.pt') +# model('./resources/01.png',show=True,save=True) +# model('./resources/FHC09594.jpg',show=True,save=True) +# model('./resources/02.jpg',show=True,save=True) +model('./dataset/val',save=True) \ No newline at end of file diff --git a/02.训练模型.py b/94.训练模型.py similarity index 100% rename from 02.训练模型.py rename to 94.训练模型.py diff --git a/03.检测训练模型.py b/95.检测训练模型.py similarity index 100% rename from 03.检测训练模型.py rename to 95.检测训练模型.py diff --git a/04.检测屏幕内容.py b/96.检测屏幕内容.py similarity index 90% rename from 04.检测屏幕内容.py rename to 96.检测屏幕内容.py index 032b1a1..9c26452 100644 --- a/04.检测屏幕内容.py +++ b/96.检测屏幕内容.py @@ -8,7 +8,7 @@ import numpy as np # 加载训练好的模型 # best.pt: 最佳模型,适用于生产 # last.pt: 最后一轮训练的模型,适用于继续训练 -yolo = YOLO('runs/detect/train2/weights/best.pt') +yolo = YOLO('runs/detect/train6/weights/best.pt') # 指定屏幕范围 # x,y,width,height 全屏None diff --git a/05.应用-视频马赛克.py b/97.应用-视频马赛克.py similarity index 100% rename from 05.应用-视频马赛克.py rename to 97.应用-视频马赛克.py diff --git a/dataset/classes.txt b/dataset/classes.txt deleted file mode 100644 index 5e49b80..0000000 --- a/dataset/classes.txt +++ /dev/null @@ -1 +0,0 @@ -logo \ No newline at end of file diff --git a/dataset/images/1.png b/dataset/images/1.png deleted file mode 100644 index a22cfff..0000000 Binary files a/dataset/images/1.png and /dev/null differ diff --git a/dataset/images/2.png b/dataset/images/2.png deleted file mode 100644 index 31077f1..0000000 Binary files a/dataset/images/2.png and /dev/null differ diff --git a/dataset/labels.cache b/dataset/labels.cache deleted file mode 100644 index 493724f..0000000 Binary files a/dataset/labels.cache and /dev/null differ diff --git a/dataset/labels/1.txt b/dataset/labels/1.txt deleted file mode 100644 index cd624fe..0000000 --- a/dataset/labels/1.txt +++ /dev/null @@ -1 +0,0 @@ -0 0.267786 0.495737 0.158273 0.103739 \ No newline at end of file diff --git a/dataset/labels/2.txt b/dataset/labels/2.txt deleted file mode 100644 index 7ad50a6..0000000 --- a/dataset/labels/2.txt +++ /dev/null @@ -1,2 +0,0 @@ -0 0.268585 0.198019 0.163070 0.034106 -0 0.499600 0.501421 0.001599 0.001421 \ No newline at end of file diff --git a/dataset/val.cache b/dataset/val.cache deleted file mode 100644 index 3aa6cee..0000000 Binary files a/dataset/val.cache and /dev/null differ diff --git a/dataset/val/1.png b/dataset/val/1.png deleted file mode 100644 index a22cfff..0000000 Binary files a/dataset/val/1.png and /dev/null differ diff --git a/dataset/val/2.png b/dataset/val/2.png deleted file mode 100644 index 31077f1..0000000 Binary files a/dataset/val/2.png and /dev/null differ diff --git a/requirements.txt b/requirements.txt index c6856f7..70e3f51 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,4 @@ -ultralytics \ No newline at end of file +ultralytics~=8.3.198 +opencv-python~=4.12.0.88 +numpy~=2.0.2 +pyautogui \ No newline at end of file