マガジンのカバー画像

ObjectDetection

7
運営しているクリエイター

#Machine_Learning

Mask RCNNで画像をマスクする

import torchimport cv2import torchvisionfrom PIL import Imageimport torchvision.transforms as Timport numpy as npimport randomimport matplotlib.pyplot as pltimport osCOCO_INSTANCE_CATEGORY_NAMES = [

もっとみる

YOLO Training Dataset

YOLOで使用するデータセットにおける、アノテーションの記載形式を示す。1行につき1つの物体を示しており、この例では3つの物体について記載されている。各行はスペース区切りで左から、物体のカテゴリID、物体中心の座標xおよびy、物体の高さhおよび幅wである。Fig. 2. はこのアノテーションに対応した画像ファイルであり、さらにFig. 1. の1行目に記載されたPottedplantの情報との対応

もっとみる