Tập dữ liệu Open Images V7
Open Images V7 là một tập dữ liệu đa năng và mở rộng được ủng hộ bởi Google . Nhằm mục đích thúc đẩy nghiên cứu trong lĩnh vực thị giác máy tính , nó tự hào có một bộ sưu tập hình ảnh khổng lồ được chú thích bằng vô số dữ liệu, bao gồm nhãn cấp độ hình ảnh, hộp giới hạn đối tượng, mặt nạ phân đoạn đối tượng, mối quan hệ trực quan và tường thuật cục bộ.
Xem: Phát hiện đối tượng sử dụng Mô hình được đào tạo trước OpenImagesV7
Open Images V7 Pretrained Model
| Mẫu | kích thước (điểm ảnh) | bản đồVal 50-95 | Tốc độ CPU ONNX (Cô) | Tốc độ A100 TensorRT (Cô) | Params (M) | Thất bại (B) |
|---|---|---|---|---|---|---|
| YOLOv8n | 640 | 18.4 | 142.4 | 1.21 | 3.5 | 10.5 |
| YOLOv8s | 640 | 27.7 | 183.1 | 1.40 | 11.4 | 29.7 |
| YOLOv8m | 640 | 33.6 | 408.5 | 2.26 | 26.2 | 80.6 |
| YOLOv8l | 640 | 34.9 | 596.9 | 2.43 | 44.1 | 167.4 |
| YOLOv8x | 640 | 36.3 | 860.6 | 3.56 | 68.7 | 260.6 |
Bạn có thể sử dụng những dữ liệu được đào tạo trước này để suy luận hoặc tinh chỉnh như sau.
Ví dụ sử dụng mô hình được đào tạo trước
Các tính năng chính
- Bao gồm ~ 9M hình ảnh được chú thích theo nhiều cách khác nhau để phù hợp với nhiều tác vụ thị giác máy tính.
- Chứa 16 triệu hộp giới hạn đáng kinh ngạc trên 600 lớp đối tượng trong 1,9 triệu hình ảnh. Các hộp này chủ yếu được vẽ bằng tay bởi các chuyên gia đảm bảo độ chính xác cao.
- Chú thích mối quan hệ trực quan tổng cộng 3,3 triệu có sẵn, chi tiết 1.466 bộ ba mối quan hệ duy nhất, thuộc tính đối tượng và hoạt động của con người.
- V5 đã giới thiệu mặt nạ phân đoạn cho 2,8 triệu đối tượng trên 350 lớp.
- V6 đã giới thiệu 675k câu chuyện được bản địa hóa kết hợp các dấu vết giọng nói, văn bản và chuột làm nổi bật các đối tượng được mô tả.
- V7 giới thiệu 66,4 triệu nhãn cấp điểm trên 1,4 triệu hình ảnh, trải dài trên 5.827 lớp.
- Bao gồm 61,4 triệu nhãn cấp hình ảnh trên một bộ đa dạng gồm 20.638 lớp.
- Cung cấp nền tảng thống nhất để phân loại hình ảnh, phát hiện đối tượng, phát hiện mối quan hệ, phân đoạn trường hợp và mô tả hình ảnh đa phương thức.
Cấu trúc tập dữ liệu
Open Images V7 được cấu trúc thành nhiều thành phần phục vụ cho các thách thức thị giác máy tính khác nhau:
- Hình ảnh: Khoảng 9 triệu hình ảnh, thường thể hiện các cảnh phức tạp với trung bình 8,3 đối tượng trên mỗi hình ảnh.
- Hộp giới hạn: Hơn 16 triệu hộp phân định ranh giới các đối tượng trên 600 danh mục.
- Mặt nạ phân đoạn: Chúng mô tả chi tiết ranh giới chính xác của 2,8 triệu đối tượng trên 350 lớp.
- Mối quan hệ trực quan: 3,3 triệu chú thích cho biết mối quan hệ, thuộc tính và hành động của đối tượng.
- Tường thuật được bản địa hóa: 675k mô tả kết hợp dấu vết giọng nói, văn bản và chuột.
- Nhãn cấp điểm : 66,4 triệu nhãn trên 1,4 triệu hình ảnh, phù hợp với phân đoạn ngữ nghĩa không/ít ảnh.
Ứng dụng
Open Images V7 là nền tảng để đào tạo và đánh giá các mô hình hiện đại trong các tác vụ thị giác máy tính khác nhau. Phạm vi rộng và chú thích chất lượng cao của tập dữ liệu làm cho nó không thể thiếu đối với các nhà nghiên cứu và nhà phát triển chuyên về thị giác máy tính.
Tập dữ liệu YAML
Thông thường, các bộ dữ liệu đi kèm với tệp YAML (Yet Another Markup Language) mô tả cấu hình của tập dữ liệu. Đối với trường hợp của Open Images V7, một giả thuyết OpenImagesV7.yaml có thể tồn tại. Để có đường dẫn và cấu hình chính xác, người ta nên tham khảo kho lưu trữ hoặc tài liệu chính thức của tập dữ liệu.
OpenImagesV7.yaml
# Ultralytics YOLO 🚀, AGPL-3.0 license
# Open Images v7 dataset https://storage.googleapis.com/openimages/web/index.html by Google
# Documentation: https://docs.ultralytics.com/datasets/detect/open-images-v7/
# Example usage: yolo train data=open-images-v7.yaml
# parent
# ├── ultralytics
# └── datasets
# └── open-images-v7 ← downloads here (561 GB)
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: ../datasets/open-images-v7 # dataset root dir
train: images/train # train images (relative to 'path') 1743042 images
val: images/val # val images (relative to 'path') 41620 images
test: # test images (optional)
# Classes
names:
0: Accordion
1: Adhesive tape
2: Aircraft
3: Airplane
4: Alarm clock
5: Alpaca
6: Ambulance
7: Animal
8: Ant
9: Antelope
10: Apple
11: Armadillo
12: Artichoke
13: Auto part
14: Axe
15: Backpack
16: Bagel
17: Baked goods
18: Balance beam
19: Ball
20: Balloon
21: Banana
22: Band-aid
23: Banjo
24: Barge
25: Barrel
26: Baseball bat
27: Baseball glove
28: Bat (Animal)
29: Bathroom accessory
30: Bathroom cabinet
31: Bathtub
32: Beaker
33: Bear
34: Bed
35: Bee
36: Beehive
37: Beer
38: Beetle
39: Bell pepper
40: Belt
41: Bench
42: Bicycle
43: Bicycle helmet
44: Bicycle wheel
45: Bidet
46: Billboard
47: Billiard table
48: Binoculars
49: Bird
50: Blender
51: Blue jay
52: Boat
53: Bomb
54: Book
55: Bookcase
56: Boot
57: Bottle
58: Bottle opener
59: Bow and arrow
60: Bowl
61: Bowling equipment
62: Box
63: Boy
64: Brassiere
65: Bread
66: Briefcase
67: Broccoli
68: Bronze sculpture
69: Brown bear
70: Building
71: Bull
72: Burrito
73: Bus
74: Bust
75: Butterfly
76: Cabbage
77: Cabinetry
78: Cake
79: Cake stand
80: Calculator
81: Camel
82: Camera
83: Can opener
84: Canary
85: Candle
86: Candy
87: Cannon
88: Canoe
89: Cantaloupe
90: Car
91: Carnivore
92: Carrot
93: Cart
94: Cassette deck
95: Castle
96: Cat
97: Cat furniture
98: Caterpillar
99: Cattle
100: Ceiling fan
101: Cello
102: Centipede
103: Chainsaw
104: Chair
105: Cheese
106: Cheetah
107: Chest of drawers
108: Chicken
109: Chime
110: Chisel
111: Chopsticks
112: Christmas tree
113: Clock
114: Closet
115: Clothing
116: Coat
117: Cocktail
118: Cocktail shaker
119: Coconut
120: Coffee
121: Coffee cup
122: Coffee table
123: Coffeemaker
124: Coin
125: Common fig
126: Common sunflower
127: Computer keyboard
128: Computer monitor
129: Computer mouse
130: Container
131: Convenience store
132: Cookie
133: Cooking spray
134: Corded phone
135: Cosmetics
136: Couch
137: Countertop
138: Cowboy hat
139: Crab
140: Cream
141: Cricket ball
142: Crocodile
143: Croissant
144: Crown
145: Crutch
146: Cucumber
147: Cupboard
148: Curtain
149: Cutting board
150: Dagger
151: Dairy Product
152: Deer
153: Desk
154: Dessert
155: Diaper
156: Dice
157: Digital clock
158: Dinosaur
159: Dishwasher
160: Dog
161: Dog bed
162: Doll
163: Dolphin
164: Door
165: Door handle
166: Doughnut
167: Dragonfly
168: Drawer
169: Dress
170: Drill (Tool)
171: Drink
172: Drinking straw
173: Drum
174: Duck
175: Dumbbell
176: Eagle
177: Earrings
178: Egg (Food)
179: Elephant
180: Envelope
181: Eraser
182: Face powder
183: Facial tissue holder
184: Falcon
185: Fashion accessory
186: Fast food
187: Fax
188: Fedora
189: Filing cabinet
190: Fire hydrant
191: Fireplace
192: Fish
193: Flag
194: Flashlight
195: Flower
196: Flowerpot
197: Flute
198: Flying disc
199: Food
200: Food processor
201: Football
202: Football helmet
203: Footwear
204: Fork
205: Fountain
206: Fox
207: French fries
208: French horn
209: Frog
210: Fruit
211: Frying pan
212: Furniture
213: Garden Asparagus
214: Gas stove
215: Giraffe
216: Girl
217: Glasses
218: Glove
219: Goat
220: Goggles
221: Goldfish
222: Golf ball
223: Golf cart
224: Gondola
225: Goose
226: Grape
227: Grapefruit
228: Grinder
229: Guacamole
230: Guitar
231: Hair dryer
232: Hair spray
233: Hamburger
234: Hammer
235: Hamster
236: Hand dryer
237: Handbag
238: Handgun
239: Harbor seal
240: Harmonica
241: Harp
242: Harpsichord
243: Hat
244: Headphones
245: Heater
246: Hedgehog
247: Helicopter
248: Helmet
249: High heels
250: Hiking equipment
251: Hippopotamus
252: Home appliance
253: Honeycomb
254: Horizontal bar
255: Horse
256: Hot dog
257: House
258: Houseplant
259: Human arm
260: Human beard
261: Human body
262: Human ear
263: Human eye
264: Human face
265: Human foot
266: Human hair
267: Human hand
268: Human head
269: Human leg
270: Human mouth
271: Human nose
272: Humidifier
273: Ice cream
274: Indoor rower
275: Infant bed
276: Insect
277: Invertebrate
278: Ipod
279: Isopod
280: Jacket
281: Jacuzzi
282: Jaguar (Animal)
283: Jeans
284: Jellyfish
285: Jet ski
286: Jug
287: Juice
288: Kangaroo
289: Kettle
290: Kitchen & dining room table
291: Kitchen appliance
292: Kitchen knife
293: Kitchen utensil
294: Kitchenware
295: Kite
296: Knife
297: Koala
298: Ladder
299: Ladle
300: Ladybug
301: Lamp
302: Land vehicle
303: Lantern
304: Laptop
305: Lavender (Plant)
306: Lemon
307: Leopard
308: Light bulb
309: Light switch
310: Lighthouse
311: Lily
312: Limousine
313: Lion
314: Lipstick
315: Lizard
316: Lobster
317: Loveseat
318: Luggage and bags
319: Lynx
320: Magpie
321: Mammal
322: Man
323: Mango
324: Maple
325: Maracas
326: Marine invertebrates
327: Marine mammal
328: Measuring cup
329: Mechanical fan
330: Medical equipment
331: Microphone
332: Microwave oven
333: Milk
334: Miniskirt
335: Mirror
336: Missile
337: Mixer
338: Mixing bowl
339: Mobile phone
340: Monkey
341: Moths and butterflies
342: Motorcycle
343: Mouse
344: Muffin
345: Mug
346: Mule
347: Mushroom
348: Musical instrument
349: Musical keyboard
350: Nail (Construction)
351: Necklace
352: Nightstand
353: Oboe
354: Office building
355: Office supplies
356: Orange
357: Organ (Musical Instrument)
358: Ostrich
359: Otter
360: Oven
361: Owl
362: Oyster
363: Paddle
364: Palm tree
365: Pancake
366: Panda
367: Paper cutter
368: Paper towel
369: Parachute
370: Parking meter
371: Parrot
372: Pasta
373: Pastry
374: Peach
375: Pear
376: Pen
377: Pencil case
378: Pencil sharpener
379: Penguin
380: Perfume
381: Person
382: Personal care
383: Personal flotation device
384: Piano
385: Picnic basket
386: Picture frame
387: Pig
388: Pillow
389: Pineapple
390: Pitcher (Container)
391: Pizza
392: Pizza cutter
393: Plant
394: Plastic bag
395: Plate
396: Platter
397: Plumbing fixture
398: Polar bear
399: Pomegranate
400: Popcorn
401: Porch
402: Porcupine
403: Poster
404: Potato
405: Power plugs and sockets
406: Pressure cooker
407: Pretzel
408: Printer
409: Pumpkin
410: Punching bag
411: Rabbit
412: Raccoon
413: Racket
414: Radish
415: Ratchet (Device)
416: Raven
417: Rays and skates
418: Red panda
419: Refrigerator
420: Remote control
421: Reptile
422: Rhinoceros
423: Rifle
424: Ring binder
425: Rocket
426: Roller skates
427: Rose
428: Rugby ball
429: Ruler
430: Salad
431: Salt and pepper shakers
432: Sandal
433: Sandwich
434: Saucer
435: Saxophone
436: Scale
437: Scarf
438: Scissors
439: Scoreboard
440: Scorpion
441: Screwdriver
442: Sculpture
443: Sea lion
444: Sea turtle
445: Seafood
446: Seahorse
447: Seat belt
448: Segway
449: Serving tray
450: Sewing machine
451: Shark
452: Sheep
453: Shelf
454: Shellfish
455: Shirt
456: Shorts
457: Shotgun
458: Shower
459: Shrimp
460: Sink
461: Skateboard
462: Ski
463: Skirt
464: Skull
465: Skunk
466: Skyscraper
467: Slow cooker
468: Snack
469: Snail
470: Snake
471: Snowboard
472: Snowman
473: Snowmobile
474: Snowplow
475: Soap dispenser
476: Sock
477: Sofa bed
478: Sombrero
479: Sparrow
480: Spatula
481: Spice rack
482: Spider
483: Spoon
484: Sports equipment
485: Sports uniform
486: Squash (Plant)
487: Squid
488: Squirrel
489: Stairs
490: Stapler
491: Starfish
492: Stationary bicycle
493: Stethoscope
494: Stool
495: Stop sign
496: Strawberry
497: Street light
498: Stretcher
499: Studio couch
500: Submarine
501: Submarine sandwich
502: Suit
503: Suitcase
504: Sun hat
505: Sunglasses
506: Surfboard
507: Sushi
508: Swan
509: Swim cap
510: Swimming pool
511: Swimwear
512: Sword
513: Syringe
514: Table
515: Table tennis racket
516: Tablet computer
517: Tableware
518: Taco
519: Tank
520: Tap
521: Tart
522: Taxi
523: Tea
524: Teapot
525: Teddy bear
526: Telephone
527: Television
528: Tennis ball
529: Tennis racket
530: Tent
531: Tiara
532: Tick
533: Tie
534: Tiger
535: Tin can
536: Tire
537: Toaster
538: Toilet
539: Toilet paper
540: Tomato
541: Tool
542: Toothbrush
543: Torch
544: Tortoise
545: Towel
546: Tower
547: Toy
548: Traffic light
549: Traffic sign
550: Train
551: Training bench
552: Treadmill
553: Tree
554: Tree house
555: Tripod
556: Trombone
557: Trousers
558: Truck
559: Trumpet
560: Turkey
561: Turtle
562: Umbrella
563: Unicycle
564: Van
565: Vase
566: Vegetable
567: Vehicle
568: Vehicle registration plate
569: Violin
570: Volleyball (Ball)
571: Waffle
572: Waffle iron
573: Wall clock
574: Wardrobe
575: Washing machine
576: Waste container
577: Watch
578: Watercraft
579: Watermelon
580: Weapon
581: Whale
582: Wheel
583: Wheelchair
584: Whisk
585: Whiteboard
586: Willow
587: Window
588: Window blind
589: Wine
590: Wine glass
591: Wine rack
592: Winter melon
593: Wok
594: Woman
595: Wood-burning stove
596: Woodpecker
597: Worm
598: Wrench
599: Zebra
600: Zucchini
# Download script/URL (optional) ---------------------------------------------------------------------------------------
download: |
from ultralytics.utils import LOGGER, SETTINGS, Path, is_ubuntu, get_ubuntu_version
from ultralytics.utils.checks import check_requirements, check_version
check_requirements('fiftyone')
if is_ubuntu() and check_version(get_ubuntu_version(), '>=22.04'):
# Ubuntu>=22.04 patch https://github.com/voxel51/fiftyone/issues/2961#issuecomment-1666519347
check_requirements('fiftyone-db-ubuntu2204')
import fiftyone as fo
import fiftyone.zoo as foz
import warnings
name = 'open-images-v7'
fraction = 1.0 # fraction of full dataset to use
LOGGER.warning('WARNING ⚠️ Open Images V7 dataset requires at least **561 GB of free space. Starting download...')
for split in 'train', 'validation': # 1743042 train, 41620 val images
train = split == 'train'
# Load Open Images dataset
dataset = foz.load_zoo_dataset(name,
split=split,
label_types=['detections'],
dataset_dir=Path(SETTINGS['datasets_dir']) / 'fiftyone' / name,
max_samples=round((1743042 if train else 41620) * fraction))
# Define classes
if train:
classes = dataset.default_classes # all classes
# classes = dataset.distinct('ground_truth.detections.label') # only observed classes
# Export to YOLO format
with warnings.catch_warnings():
warnings.filterwarnings("ignore", category=UserWarning, module="fiftyone.utils.yolo")
dataset.export(export_dir=str(Path(SETTINGS['datasets_dir']) / name),
dataset_type=fo.types.YOLOv5Dataset,
label_field='ground_truth',
split='val' if split == 'validation' else split,
classes=classes,
overwrite=train)
Sử dụng
Để đào tạo mô hình YOLO11n trên tập dữ liệu Open Images V7 cho 100 kỷ nguyên với kích thước hình ảnh là 640, bạn có thể sử dụng các đoạn mã sau. Để biết danh sách đầy đủ các đối số khả dụng, hãy tham khảo trang Đào tạo mô hình.
Cảnh báo
Bộ dữ liệu Open Images V7 hoàn chỉnh bao gồm 1.743.042 hình ảnh đào tạo và 41.620 hình ảnh xác thực, yêu cầu khoảng 561 GB dung lượng lưu trữ khi tải xuống.
Thực hiện các lệnh được cung cấp bên dưới sẽ kích hoạt tải xuống tự động tập dữ liệu đầy đủ nếu nó chưa có mặt cục bộ. Trước khi chạy ví dụ dưới đây, điều quan trọng là:
- Xác minh rằng thiết bị của bạn có đủ dung lượng lưu trữ.
- Đảm bảo kết nối internet mạnh mẽ và nhanh chóng.
Ví dụ về tàu hỏa
Dữ liệu mẫu và chú thích
Minh họa của tập dữ liệu giúp cung cấp thông tin chi tiết về sự phong phú của nó:
- Open Images V7: Hình ảnh này minh họa độ sâu và chi tiết của các chú thích có sẵn, bao gồm các hộp giới hạn, mối quan hệ và mặt nạ phân đoạn.
Các nhà nghiên cứu có thể có được những hiểu biết vô giá về một loạt các thách thức thị giác máy tính mà bộ dữ liệu giải quyết, từ phát hiện đối tượng cơ bản đến xác định mối quan hệ phức tạp.
Trích dẫn và xác nhận
Đối với những người sử dụng Open Images V7 trong công việc của họ, thật thận trọng khi trích dẫn các bài báo liên quan và thừa nhận những người sáng tạo:
@article{OpenImages,
author = {Alina Kuznetsova and Hassan Rom and Neil Alldrin and Jasper Uijlings and Ivan Krasin and Jordi Pont-Tuset and Shahab Kamali and Stefan Popov and Matteo Malloci and Alexander Kolesnikov and Tom Duerig and Vittorio Ferrari},
title = {The Open Images Dataset V4: Unified image classification, object detection, and visual relationship detection at scale},
year = {2020},
journal = {IJCV}
}
Một lời cảm ơn chân thành dành cho Google Nhóm AI để tạo và duy trì tập dữ liệu Open Images V7. Để tìm hiểu sâu về tập dữ liệu và các dịch vụ của nó, hãy điều hướng đến trang web Open Images V7 chính thức.
FAQ
Tập dữ liệu Open Images V7 là gì?
Open Images V7 là một tập dữ liệu mở rộng và linh hoạt được tạo bởi Google, được thiết kế để thúc đẩy nghiên cứu về thị giác máy tính. Nó bao gồm các nhãn cấp độ hình ảnh, hộp giới hạn đối tượng, mặt nạ phân đoạn đối tượng, mối quan hệ trực quan và tường thuật được bản địa hóa, làm cho nó trở nên lý tưởng cho các tác vụ thị giác máy tính khác nhau như phát hiện đối tượng, phân đoạn và phát hiện mối quan hệ.
Làm thế nào để đào tạo mô hình YOLO11 trên tập dữ liệu Open Images V7?
Để đào tạo mô hình YOLO11 trên tập dữ liệu Open Images V7, bạn có thể sử dụng cả hai Python Và CLI lệnh. Sau đây là ví dụ về việc đào tạo mô hình YOLO11n trong 100 kỷ nguyên với kích thước hình ảnh là 640:
Ví dụ về tàu hỏa
Để biết thêm chi tiết về các đối số và thiết đặt, hãy tham khảo trang Đào tạo .
Một số tính năng chính của tập dữ liệu Open Images V7 là gì?
Bộ dữ liệu Open Images V7 bao gồm khoảng 9 triệu hình ảnh với nhiều chú thích khác nhau:
- Hộp giới hạn : 16 triệu hộp giới hạn trên 600 lớp đối tượng.
- Mặt nạ phân đoạn : Mặt nạ cho 2,8 triệu đối tượng trên 350 lớp.
- Mối quan hệ trực quan : 3,3 triệu chú thích chỉ ra mối quan hệ, thuộc tính và hành động.
- Nội dung tường thuật được bản địa hóa : 675.000 mô tả kết hợp giọng nói, văn bản và dấu vết chuột.
- Nhãn cấp điểm : 66,4 triệu nhãn trên 1,4 triệu hình ảnh.
- Nhãn cấp hình ảnh : 61,4 triệu nhãn trên 20.638 lớp.
Những mô hình được đào tạo trước nào có sẵn cho tập dữ liệu Open Images V7?
Ultralytics cung cấp một số YOLOv8 các mô hình được đào tạo trước cho tập dữ liệu Open Images V7, mỗi mô hình có kích thước và chỉ số hiệu suất khác nhau:
| Mẫu | kích thước (điểm ảnh) | bản đồVal 50-95 | Tốc độ CPU ONNX (Cô) | Tốc độ A100 TensorRT (Cô) | Params (M) | Thất bại (B) |
|---|---|---|---|---|---|---|
| YOLOv8n | 640 | 18.4 | 142.4 | 1.21 | 3.5 | 10.5 |
| YOLOv8s | 640 | 27.7 | 183.1 | 1.40 | 11.4 | 29.7 |
| YOLOv8m | 640 | 33.6 | 408.5 | 2.26 | 26.2 | 80.6 |
| YOLOv8l | 640 | 34.9 | 596.9 | 2.43 | 44.1 | 167.4 |
| YOLOv8x | 640 | 36.3 | 860.6 | 3.56 | 68.7 | 260.6 |
Tập dữ liệu Open Images V7 có thể được sử dụng cho những ứng dụng nào?
Bộ dữ liệu Open Images V7 hỗ trợ nhiều tác vụ thị giác máy tính bao gồm:
- Phân loại hình ảnh
- Phát hiện đối tượng
- Phân đoạn phiên bản
- Phát hiện mối quan hệ trực quan
- Mô tả hình ảnh đa phương thức
Các chú thích toàn diện và phạm vi rộng của nó làm cho nó phù hợp để đào tạo và đánh giá các mô hình học máy tiên tiến, như được nêu bật trong các trường hợp sử dụng thực tế được nêu chi tiết trong phần ứng dụng của chúng tôi.
