Chuyển đến nội dung

Bộ dữ liệu Objects365

Bộ dữ liệu Objects365 là một bộ dữ liệu quy mô lớn, chất lượng cao được thiết kế để thúc đẩy nghiên cứu phát hiện đối tượng, tập trung vào các đối tượng đa dạng trong tự nhiên. Được tạo bởi một nhóm các nhà nghiên cứu Megvii, bộ dữ liệu cung cấp một loạt các hình ảnh có độ phân giải cao với một tập hợp toàn diện các hộp giới hạn được chú thích bao phủ 365 danh mục đối tượng.



Xem: Cách Huấn luyện Ultralytics YOLO11 trên Bộ dữ liệu Objects365 với Ultralytics | 2M Chú thích 🚀

Các tính năng chính

  • Objects365 chứa 365 danh mục đối tượng, với 2 triệu hình ảnh và hơn 30 triệu hộp giới hạn.
  • Bộ dữ liệu bao gồm các đối tượng đa dạng trong các tình huống khác nhau, cung cấp một chuẩn mực phong phú và đầy thách thức cho các tác vụ phát hiện đối tượng.
  • Các chú thích bao gồm các hộp giới hạn cho các đối tượng, làm cho nó phù hợp để huấn luyện và đánh giá các mô hình phát hiện đối tượng.
  • Các mô hình Objects365 được huấn luyện trước vượt trội hơn đáng kể so với các mô hình ImageNet được huấn luyện trước, dẫn đến khả năng khái quát hóa tốt hơn trên nhiều tác vụ khác nhau.

Cấu trúc bộ dữ liệu

Bộ dữ liệu Objects365 được tổ chức thành một tập hợp hình ảnh duy nhất với các chú thích tương ứng:

  • Images: Bộ dữ liệu bao gồm 2 triệu hình ảnh độ phân giải cao, mỗi hình ảnh chứa nhiều đối tượng thuộc 365 danh mục.
  • Chú thích: Các hình ảnh được chú thích với hơn 30 triệu bounding box, cung cấp thông tin ground truth toàn diện cho các tác vụ phát hiện đối tượng.

Các ứng dụng

Bộ dữ liệu Objects365 được sử dụng rộng rãi để huấn luyện và đánh giá các mô hình học sâu trong các tác vụ phát hiện đối tượng. Tập hợp đa dạng các danh mục đối tượng và chú thích chất lượng cao của bộ dữ liệu này làm cho nó trở thành một nguồn tài nguyên có giá trị cho các nhà nghiên cứu và người thực hành trong lĩnh vực thị giác máy tính.

YAML bộ dữ liệu

Tệp YAML (Yet Another Markup Language) được sử dụng để xác định cấu hình bộ dữ liệu. Nó chứa thông tin về đường dẫn, các lớp và các thông tin liên quan khác của bộ dữ liệu. Đối với trường hợp bộ dữ liệu Objects365, Objects365.yaml tệp được duy trì tại https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/Objects365.yaml.

ultralytics/cfg/datasets/Objects365.yaml

# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

# Objects365 dataset https://www.objects365.org/ by Megvii
# Documentation: https://docs.ultralytics.com/datasets/detect/objects365/
# Example usage: yolo train data=Objects365.yaml
# parent
# ├── ultralytics
# └── datasets
#     └── Objects365 ← downloads here (712 GB = 367G data + 345G zips)

# 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: Objects365 # dataset root dir
train: images/train # train images (relative to 'path') 1742289 images
val: images/val # val images (relative to 'path') 80000 images
test: # test images (optional)

# Classes
names:
  0: Person
  1: Sneakers
  2: Chair
  3: Other Shoes
  4: Hat
  5: Car
  6: Lamp
  7: Glasses
  8: Bottle
  9: Desk
  10: Cup
  11: Street Lights
  12: Cabinet/shelf
  13: Handbag/Satchel
  14: Bracelet
  15: Plate
  16: Picture/Frame
  17: Helmet
  18: Book
  19: Gloves
  20: Storage box
  21: Boat
  22: Leather Shoes
  23: Flower
  24: Bench
  25: Potted Plant
  26: Bowl/Basin
  27: Flag
  28: Pillow
  29: Boots
  30: Vase
  31: Microphone
  32: Necklace
  33: Ring
  34: SUV
  35: Wine Glass
  36: Belt
  37: Monitor/TV
  38: Backpack
  39: Umbrella
  40: Traffic Light
  41: Speaker
  42: Watch
  43: Tie
  44: Trash bin Can
  45: Slippers
  46: Bicycle
  47: Stool
  48: Barrel/bucket
  49: Van
  50: Couch
  51: Sandals
  52: Basket
  53: Drum
  54: Pen/Pencil
  55: Bus
  56: Wild Bird
  57: High Heels
  58: Motorcycle
  59: Guitar
  60: Carpet
  61: Cell Phone
  62: Bread
  63: Camera
  64: Canned
  65: Truck
  66: Traffic cone
  67: Cymbal
  68: Lifesaver
  69: Towel
  70: Stuffed Toy
  71: Candle
  72: Sailboat
  73: Laptop
  74: Awning
  75: Bed
  76: Faucet
  77: Tent
  78: Horse
  79: Mirror
  80: Power outlet
  81: Sink
  82: Apple
  83: Air Conditioner
  84: Knife
  85: Hockey Stick
  86: Paddle
  87: Pickup Truck
  88: Fork
  89: Traffic Sign
  90: Balloon
  91: Tripod
  92: Dog
  93: Spoon
  94: Clock
  95: Pot
  96: Cow
  97: Cake
  98: Dining Table
  99: Sheep
  100: Hanger
  101: Blackboard/Whiteboard
  102: Napkin
  103: Other Fish
  104: Orange/Tangerine
  105: Toiletry
  106: Keyboard
  107: Tomato
  108: Lantern
  109: Machinery Vehicle
  110: Fan
  111: Green Vegetables
  112: Banana
  113: Baseball Glove
  114: Airplane
  115: Mouse
  116: Train
  117: Pumpkin
  118: Soccer
  119: Skiboard
  120: Luggage
  121: Nightstand
  122: Tea pot
  123: Telephone
  124: Trolley
  125: Head Phone
  126: Sports Car
  127: Stop Sign
  128: Dessert
  129: Scooter
  130: Stroller
  131: Crane
  132: Remote
  133: Refrigerator
  134: Oven
  135: Lemon
  136: Duck
  137: Baseball Bat
  138: Surveillance Camera
  139: Cat
  140: Jug
  141: Broccoli
  142: Piano
  143: Pizza
  144: Elephant
  145: Skateboard
  146: Surfboard
  147: Gun
  148: Skating and Skiing shoes
  149: Gas stove
  150: Donut
  151: Bow Tie
  152: Carrot
  153: Toilet
  154: Kite
  155: Strawberry
  156: Other Balls
  157: Shovel
  158: Pepper
  159: Computer Box
  160: Toilet Paper
  161: Cleaning Products
  162: Chopsticks
  163: Microwave
  164: Pigeon
  165: Baseball
  166: Cutting/chopping Board
  167: Coffee Table
  168: Side Table
  169: Scissors
  170: Marker
  171: Pie
  172: Ladder
  173: Snowboard
  174: Cookies
  175: Radiator
  176: Fire Hydrant
  177: Basketball
  178: Zebra
  179: Grape
  180: Giraffe
  181: Potato
  182: Sausage
  183: Tricycle
  184: Violin
  185: Egg
  186: Fire Extinguisher
  187: Candy
  188: Fire Truck
  189: Billiards
  190: Converter
  191: Bathtub
  192: Wheelchair
  193: Golf Club
  194: Briefcase
  195: Cucumber
  196: Cigar/Cigarette
  197: Paint Brush
  198: Pear
  199: Heavy Truck
  200: Hamburger
  201: Extractor
  202: Extension Cord
  203: Tong
  204: Tennis Racket
  205: Folder
  206: American Football
  207: earphone
  208: Mask
  209: Kettle
  210: Tennis
  211: Ship
  212: Swing
  213: Coffee Machine
  214: Slide
  215: Carriage
  216: Onion
  217: Green beans
  218: Projector
  219: Frisbee
  220: Washing Machine/Drying Machine
  221: Chicken
  222: Printer
  223: Watermelon
  224: Saxophone
  225: Tissue
  226: Toothbrush
  227: Ice cream
  228: Hot-air balloon
  229: Cello
  230: French Fries
  231: Scale
  232: Trophy
  233: Cabbage
  234: Hot dog
  235: Blender
  236: Peach
  237: Rice
  238: Wallet/Purse
  239: Volleyball
  240: Deer
  241: Goose
  242: Tape
  243: Tablet
  244: Cosmetics
  245: Trumpet
  246: Pineapple
  247: Golf Ball
  248: Ambulance
  249: Parking meter
  250: Mango
  251: Key
  252: Hurdle
  253: Fishing Rod
  254: Medal
  255: Flute
  256: Brush
  257: Penguin
  258: Megaphone
  259: Corn
  260: Lettuce
  261: Garlic
  262: Swan
  263: Helicopter
  264: Green Onion
  265: Sandwich
  266: Nuts
  267: Speed Limit Sign
  268: Induction Cooker
  269: Broom
  270: Trombone
  271: Plum
  272: Rickshaw
  273: Goldfish
  274: Kiwi fruit
  275: Router/modem
  276: Poker Card
  277: Toaster
  278: Shrimp
  279: Sushi
  280: Cheese
  281: Notepaper
  282: Cherry
  283: Pliers
  284: CD
  285: Pasta
  286: Hammer
  287: Cue
  288: Avocado
  289: Hami melon
  290: Flask
  291: Mushroom
  292: Screwdriver
  293: Soap
  294: Recorder
  295: Bear
  296: Eggplant
  297: Board Eraser
  298: Coconut
  299: Tape Measure/Ruler
  300: Pig
  301: Showerhead
  302: Globe
  303: Chips
  304: Steak
  305: Crosswalk Sign
  306: Stapler
  307: Camel
  308: Formula 1
  309: Pomegranate
  310: Dishwasher
  311: Crab
  312: Hoverboard
  313: Meatball
  314: Rice Cooker
  315: Tuba
  316: Calculator
  317: Papaya
  318: Antelope
  319: Parrot
  320: Seal
  321: Butterfly
  322: Dumbbell
  323: Donkey
  324: Lion
  325: Urinal
  326: Dolphin
  327: Electric Drill
  328: Hair Dryer
  329: Egg tart
  330: Jellyfish
  331: Treadmill
  332: Lighter
  333: Grapefruit
  334: Game board
  335: Mop
  336: Radish
  337: Baozi
  338: Target
  339: French
  340: Spring Rolls
  341: Monkey
  342: Rabbit
  343: Pencil Case
  344: Yak
  345: Red Cabbage
  346: Binoculars
  347: Asparagus
  348: Barbell
  349: Scallop
  350: Noddles
  351: Comb
  352: Dumpling
  353: Oyster
  354: Table Tennis paddle
  355: Cosmetics Brush/Eyeliner Pencil
  356: Chainsaw
  357: Eraser
  358: Lobster
  359: Durian
  360: Okra
  361: Lipstick
  362: Cosmetics Mirror
  363: Curling
  364: Table Tennis

# Download script/URL (optional) ---------------------------------------------------------------------------------------
download: |
  from pathlib import Path

  import numpy as np

  from ultralytics.utils import TQDM
  from ultralytics.utils.checks import check_requirements
  from ultralytics.utils.downloads import download
  from ultralytics.utils.ops import xyxy2xywhn

  check_requirements("faster-coco-eval")
  from faster_coco_eval import COCO

  # Make Directories
  dir = Path(yaml["path"])  # dataset root dir
  for p in "images", "labels":
      (dir / p).mkdir(parents=True, exist_ok=True)
      for q in "train", "val":
          (dir / p / q).mkdir(parents=True, exist_ok=True)

  # Train, Val Splits
  for split, patches in [("train", 50 + 1), ("val", 43 + 1)]:
      print(f"Processing {split} in {patches} patches ...")
      images, labels = dir / "images" / split, dir / "labels" / split

      # Download
      url = f"https://dorc.ks3-cn-beijing.ksyun.com/data-set/2020Objects365%E6%95%B0%E6%8D%AE%E9%9B%86/{split}/"
      if split == "train":
          download([f"{url}zhiyuan_objv2_{split}.tar.gz"], dir=dir)  # annotations json
          download([f"{url}patch{i}.tar.gz" for i in range(patches)], dir=images, curl=True, threads=8)
      elif split == "val":
          download([f"{url}zhiyuan_objv2_{split}.json"], dir=dir)  # annotations json
          download([f"{url}images/v1/patch{i}.tar.gz" for i in range(15 + 1)], dir=images, curl=True, threads=8)
          download([f"{url}images/v2/patch{i}.tar.gz" for i in range(16, patches)], dir=images, curl=True, threads=8)

      # Move
      for f in TQDM(images.rglob("*.jpg"), desc=f"Moving {split} images"):
          f.rename(images / f.name)  # move to /images/{split}

      # Labels
      coco = COCO(dir / f"zhiyuan_objv2_{split}.json")
      names = [x["name"] for x in coco.loadCats(coco.getCatIds())]
      for cid, cat in enumerate(names):
          catIds = coco.getCatIds(catNms=[cat])
          imgIds = coco.getImgIds(catIds=catIds)
          for im in TQDM(coco.loadImgs(imgIds), desc=f"Class {cid + 1}/{len(names)} {cat}"):
              width, height = im["width"], im["height"]
              path = Path(im["file_name"])  # image filename
              try:
                  with open(labels / path.with_suffix(".txt").name, "a", encoding="utf-8") as file:
                      annIds = coco.getAnnIds(imgIds=im["id"], catIds=catIds, iscrowd=None)
                      for a in coco.loadAnns(annIds):
                          x, y, w, h = a["bbox"]  # bounding box in xywh (xy top-left corner)
                          xyxy = np.array([x, y, x + w, y + h])[None]  # pixels(1,4)
                          x, y, w, h = xyxy2xywhn(xyxy, w=width, h=height, clip=True)[0]  # normalized and clipped
                          file.write(f"{cid} {x:.5f} {y:.5f} {w:.5f} {h:.5f}\n")
              except Exception as e:
                  print(e)

Cách sử dụng

Để huấn luyện mô hình YOLO11n trên bộ dữ liệu Objects365 trong 100 epochs với kích thước hình ảnh là 640, bạn có thể sử dụng các đoạn mã sau. Để có danh sách đầy đủ các đối số có sẵn, hãy tham khảo trang Huấn luyện mô hình.

Ví dụ huấn luyện

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="Objects365.yaml", epochs=100, imgsz=640)
# Start training from a pretrained *.pt model
yolo detect train data=Objects365.yaml model=yolo11n.pt epochs=100 imgsz=640

Dữ liệu mẫu và Chú thích

Bộ dữ liệu Objects365 chứa một tập hợp đa dạng các hình ảnh có độ phân giải cao với các đối tượng từ 365 danh mục, cung cấp bối cảnh phong phú cho các tác vụ phát hiện đối tượng. Dưới đây là một số ví dụ về hình ảnh trong bộ dữ liệu:

Ảnh mẫu bộ dữ liệu

  • Objects365: Hình ảnh này minh họa một ví dụ về phát hiện đối tượng, trong đó các đối tượng được chú thích bằng các hộp giới hạn. Bộ dữ liệu cung cấp một loạt các hình ảnh để tạo điều kiện phát triển các mô hình cho tác vụ này.

Ví dụ này thể hiện sự đa dạng và phức tạp của dữ liệu trong bộ dữ liệu Objects365, đồng thời nhấn mạnh tầm quan trọng của việc phát hiện đối tượng chính xác cho các ứng dụng thị giác máy tính.

Trích dẫn và Lời cảm ơn

Nếu bạn sử dụng bộ dữ liệu Objects365 trong công việc nghiên cứu hoặc phát triển của mình, vui lòng trích dẫn bài báo sau:

@inproceedings{shao2019objects365,
  title={Objects365: A Large-scale, High-quality Dataset for Object Detection},
  author={Shao, Shuai and Li, Zeming and Zhang, Tianyuan and Peng, Chao and Yu, Gang and Li, Jing and Zhang, Xiangyu and Sun, Jian},
  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
  pages={8425--8434},
  year={2019}
}

Chúng tôi xin ghi nhận công lao của nhóm các nhà nghiên cứu đã tạo ra và duy trì bộ dữ liệu Objects365 như một nguồn tài nguyên giá trị cho cộng đồng nghiên cứu thị giác máy tính. Để biết thêm thông tin về bộ dữ liệu Objects365 và những người tạo ra nó, hãy truy cập trang web bộ dữ liệu Objects365.

Câu hỏi thường gặp

Bộ dữ liệu Objects365 được sử dụng để làm gì?

Bộ dữ liệu Objects365 được thiết kế cho các tác vụ phát hiện đối tượng trong học máy và thị giác máy tính. Nó cung cấp một bộ dữ liệu quy mô lớn, chất lượng cao với 2 triệu hình ảnh được chú thích và 30 triệu bounding box trên 365 danh mục. Việc tận dụng một bộ dữ liệu đa dạng như vậy giúp cải thiện hiệu suất và khả năng tổng quát hóa của các mô hình phát hiện đối tượng, làm cho nó vô giá cho nghiên cứu và phát triển trong lĩnh vực này.

Làm cách nào để huấn luyện mô hình YOLO11 trên bộ dữ liệu Objects365?

Để huấn luyện mô hình YOLO11n bằng bộ dữ liệu Objects365 trong 100 epochs với kích thước ảnh là 640, hãy làm theo các hướng dẫn sau:

Ví dụ huấn luyện

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="Objects365.yaml", epochs=100, imgsz=640)
# Start training from a pretrained *.pt model
yolo detect train data=Objects365.yaml model=yolo11n.pt epochs=100 imgsz=640

Tham khảo trang Huấn luyện để có danh sách đầy đủ các đối số có sẵn.

Tại sao tôi nên sử dụng bộ dữ liệu Objects365 cho các dự án phát hiện đối tượng của mình?

Bộ dữ liệu Objects365 cung cấp một số lợi thế cho các tác vụ phát hiện đối tượng:

  1. Tính đa dạng: Nó bao gồm 2 triệu hình ảnh với các đối tượng trong các tình huống đa dạng, bao gồm 365 danh mục.
  2. Chú thích chất lượng cao: Hơn 30 triệu hộp giới hạn cung cấp dữ liệu ground truth toàn diện.
  3. Hiệu suất: Các mô hình được huấn luyện trước trên Objects365 vượt trội hơn đáng kể so với các mô hình được huấn luyện trên các bộ dữ liệu như ImageNet, dẫn đến khả năng tổng quát hóa tốt hơn.

Tôi có thể tìm thấy tệp cấu hình YAML cho bộ dữ liệu Objects365 ở đâu?

Tệp cấu hình YAML cho tập dữ liệu Objects365 có sẵn tại Objects365.yaml. Tệp này chứa thông tin cần thiết như đường dẫn tập dữ liệu và nhãn lớp, rất quan trọng để thiết lập môi trường huấn luyện của bạn.

Cấu trúc bộ dữ liệu của Objects365 nâng cao mô hình phát hiện đối tượng như thế nào?

Bộ dữ liệu Objects365 được tổ chức với 2 triệu hình ảnh có độ phân giải cao và chú thích toàn diện của hơn 30 triệu hộp giới hạn. Cấu trúc này đảm bảo một bộ dữ liệu mạnh mẽ để đào tạo các mô hình học sâu trong phát hiện đối tượng, cung cấp nhiều đối tượng và kịch bản khác nhau. Sự đa dạng và khối lượng như vậy giúp phát triển các mô hình chính xác hơn và có khả năng khái quát hóa tốt cho các ứng dụng thực tế. Để biết thêm chi tiết về cấu trúc bộ dữ liệu, hãy tham khảo phần Dataset YAML.



📅 Đã tạo 1 năm trước ✏️ Cập nhật 2 tháng trước

Bình luận