Skip to content

Commit 4cb3273

Browse files
committed
* update make datasets
1 parent 6b42945 commit 4cb3273

File tree

2 files changed

+23
-4
lines changed

2 files changed

+23
-4
lines changed

main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from dataset_tools.jade_create_object_dection_datasets import CreateYearsDatasets
1212

1313
def test_create_paddle_years_datasets(args):
14-
CreateYearsDatasets(args.input_dataset_dir,None,save_path=args.save_dataset_dir,rate=0.8)
14+
CreateYearsDatasets(args.input_dataset_dir,None,save_path=args.save_dataset_dir,rate=0.9)
1515

1616
if __name__ == '__main__':
1717
import argparse
@@ -23,9 +23,9 @@ def test_create_paddle_years_datasets(args):
2323
args = parser.parse_args()
2424
print(list(args.voc_labels))
2525
if args.dataset_type == "paddle_detection":
26-
CreateYearsDatasets(args.input_dataset_dir, None, save_path=args.save_dataset_dir, rate=0.8)
26+
CreateYearsDatasets(args.input_dataset_dir, None, save_path=args.save_dataset_dir, rate=0.9)
2727
elif args.dataset_type == "yolo_detection":
28-
CreateDarknetVocDatasets(args.input_dataset_dir, save_path=args.save_dataset_dir, rate=0.8, VOC_CLASSES=args.voc_labels)
28+
CreateDarknetVocDatasets(args.input_dataset_dir, save_path=args.save_dataset_dir, rate=0.9, VOC_CLASSES=args.voc_labels)
2929

3030
#removeNolabelDatasets(r"F:\数据集\关键点检测数据集\定制版箱号关键点数据集\2022-03-09")
3131
#create_text_detection_datasets(r"F:\数据集\关键点检测数据集\定制版箱号关键点数据集",r"E:\Data\字符检测识别数据集\定制版箱号关键点数据集",0.95)

test.py

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,8 @@
77
# @Software : Samples
88
# @Desc :
99
from dataset_tools.jade_create_object_dection_datasets import *
10+
from opencv_tools import ReadChinesePath
11+
import cv2
1012

1113

1214
def testCreateYearsDatasets():
@@ -21,6 +23,23 @@ def testCreateYearsDarknetVocDatasets():
2123
VOC_CLASSES = ["FRONTEND","DOOREND","UPEND","slide","bromine_tank"]
2224
CreateYearsDarknetVocDatasets(r"F:\数据集\VOC数据集\箱门检测数据集\ContainVOC", save_path=r"E:\Data\VOC数据集\箱门检测数据集\ContainerVOCDarknet",VOC_CLASSES=VOC_CLASSES)
2325

26+
def image_to_video():
27+
image = ReadChinesePath(r"C:\Users\Administrator\Desktop\back_2025-05-07-15-40-44-701.jpg")
28+
height, width, _ = image.shape
29+
video_writer = cv2.VideoWriter(
30+
r"C:\Users\Administrator\Desktop\output_video.avi",
31+
cv2.VideoWriter_fourcc(*"XVID"),
32+
30, # Frame rate
33+
(width, height)
34+
)
35+
36+
index = 0
37+
while index < 1000:
38+
video_writer.write(image)
39+
index += 1
40+
41+
video_writer.release()
42+
2443

2544
if __name__ == '__main__':
26-
testCreateYearsDatasets()
45+
image_to_video()

0 commit comments

Comments
 (0)