Skip to content
@Airspace-Explorer

Airspace-Explorer

Detection and Tracking of abnormal objects in the air in various weather environments

ํŒ€๋ช…: Airspace-Explorer

  • ์ง€๋„ ๊ต์ˆ˜: ์ด๋Œ€ํ˜ธ
  • ํŒ€์›:
    ์†Œํ”„ํŠธ์›จ์–ด์œตํ•ฉํ•™๊ณผ ๊ตฌํ˜„์„œ 2018102091 (PM)
    ์†Œํ”„ํŠธ์›จ์–ด์œตํ•ฉํ•™๊ณผ ๊น€๋ฏผํ™˜ 2019102081
    ์†Œํ”„ํŠธ์›จ์–ด์œตํ•ฉํ•™๊ณผ ์ด์ •์› 2020110480

1.๊ฐœ์š”

[์กฐ๋ฅ˜ ์ถฉ๋Œ]
์šฐ๋ฆฌ ๋ง๋กœ '์กฐ๋ฅ˜ ์ถฉ๋Œ'์ด๋ผ๊ณ  ๋ถˆ๋ฆฌ๋Š” '๋ฒ„๋“œ ์ŠคํŠธ๋ผ์ดํฌ(Bird Strike)'๋Š” ์กฐ๋ฅ˜๊ฐ€ ๋น„ํ–‰๊ธฐ์— ๋ถ€๋”ชํžˆ๊ฑฐ๋‚˜ ์—”์ง„ ์†์— ๋นจ๋ ค ๋“ค์–ด๊ฐ€๋Š” ํ˜„์ƒ์„ ๋งํ•œ๋‹ค. ์ฃผ๋กœ ๊ณตํ•ญ ๋ถ€๊ทผ, ๊ทธ๋ฆฌ๊ณ  ์ด์ฐฉ๋ฅ™ ์‹œ ์ฃผ๋กœ ๋ฐœ์ƒํ•˜๋Š”๋ฐ, ์šฐ๋ฆฌ๋‚˜๋ผ์—์„œ๋Š” ์กฐ๋ฅ˜์ถฉ๋Œ ์‚ฌ๊ณ ๊ฐ€ ๋งค๋…„ 100~200๊ฑด ์ด์ƒ ๋ฐœ์ƒ ํ•˜๊ณ  ์žˆ์œผ๋ฉฐ ์ง€๋‚œํ•ด ๋ฏธ๊ตญ ์—ฐ๋ฐฉํ•ญ๊ณต์ฒญ์—์„œ๋Š” 1๋งŒ 7์ฒœ ๊ฑด์ด ๋„˜๋Š” ์‹ ๊ณ ๊ฐ€ ์ ‘์ˆ˜ ๋˜์—ˆ๋‹ค. ์‹ค์ œ๋กœ 1.8kg์˜ ์ƒˆ๊ฐ€ ์‹œ์† 960km๋กœ ๋น„ํ–‰ํ•˜๋Š” ํ•ญ๊ณต๊ธฐ์™€ ๋ถ€๋”ช์น˜๋ฉด 64t ๋ฌด๊ฒŒ์˜ ์ถฉ๊ฒฉ์ด ๋ฐœ์ƒํ•˜๋ฉฐ, ์ „ ์„ธ๊ณ„์  ํ”ผํ•ด๊ทœ๋ชจ๋Š” ์—ฐ๊ฐ„ ์•ฝ 1์กฐ์›์œผ๋กœ ์ถ”์ •๋œ๋‹ค. ์ตœ๊ทผ 5๋…„๊ฐ„ ํ•ญ๊ณต๊ธฐ-์กฐ๋ฅ˜๊ฐ„ ์ถฉ๋Œ์€ ์ฃผ๋กœ ๊ณตํ•ญ๊ตฌ์—ญ์—์„œ ๋ฐœ์ƒํ•˜๊ณ  ์žˆ์œผ๋ฉฐ, ์ด๋ฅผ ์˜ˆ๋ฐฉํ•˜๊ธฐ ์œ„ํ•ด ๊ณตํ•ญ์—์„œ๋Š” ์‚ฌ๊ฒฉํŒ€ ์šด์˜, ์ฒœ์ ๋ฅ˜ ์‚ฌ์œก์„ ํ†ตํ•ด ๋…ธ๋ ฅํ•˜๊ณ  ์žˆ์ง€๋งŒ ํšจ๊ณผ๊ฐ€ ๋ฏธ๋น„ํ•œ ์ƒํ™ฉ์ด๋‹ค. ๊ทธ๋ž˜์„œ ๋ณธํŒ€์€ ์ตœ์†Œ ์ˆ˜์‹ญ๋ช…์—์„œ ๋งŽ๊ฒŒ๋Š” ์ˆ˜๋ฐฑ๋ช… ์ด์ƒ์˜ ์‚ฌ๋ง์ž๋ฅผ ๋ฐœ์ƒ์‹œํ‚ค๋Š” ๋ฒ„๋“œ ์ŠคํŠธ๋ผ์ดํฌ ๋ฐฉ์ง€๋ฅผ ์œ„ํ•ด Faster-RCNN, YOLOF, SSD ๊ธฐ๋ฐ˜์˜ ์กฐ๋ฅ˜๋ฅผ ์‹ค์‹œ๊ฐ„์œผ๋กœ Detection ํ•˜๋Š” ๋ชจ๋ธ์„ ํ•™์Šต์„ ํ†ตํ•ด ๊ตฌ์ถ•ํ•˜์—ฌ ๋‹ค์–‘ํ•œ ๊ธฐ์ƒ ํ™˜๊ฒฝ์—์„œ ๋น„ํ–‰์ค‘์ธ ์ƒ๊ณต๋ฌผ์ฒด(์กฐ๋ฅ˜)๋ฅผ ์‹ค์‹œ๊ฐ„์œผ๋กœ ํƒ์ง€ํ•  ๊ฒƒ์ด๋ฉฐ, ๋˜ํ•œ Detection ๋ชจ๋ธ๊ณผ ReID ๋ชจ๋ธ์„ ๊ฒฐํ•ฉํ•˜์—ฌ DeepSORT ๋ชจ๋ธ์„ ๊ตฌ์ถ•ํ•œ ๋’ค ์‹ค์‹œ๊ฐ„์œผ๋กœ ์ƒ๊ณต ๋ฌผ์ฒด(์กฐ๋ฅ˜)๋ฅผ ์ถ”์  ํ•  ๊ฒƒ์ด๋‹ค.

[์ ์„ฑ๊ตญ ๋ฌด์ธ ํ•ญ๊ณต๊ธฐ์˜ ์˜๊ณต ์นจ๋ฒ”]
์ตœ๊ทผ ๋ถํ•œ์˜ ๋ฌด์ธ ํ•ญ๊ณต๊ธฐ์˜ ์˜๊ณต ์นจ์ž…์ด ๋ฌธ์ œ๋กœ ๋– ์˜ค๋ฅด๊ณ  ์žˆ๋‹ค. ์ ์˜ ๊ณ ์ •์ต๊ธฐ์™€ ํšŒ์ „์ต๊ธฐ, ์ค‘๋Œ€ํ˜• ๋ฌด์ธ๊ธฐ์˜ ๊ฒฝ์šฐ ํƒ์ง€ ๋ฐ ์‹๋ณ„์ด ์‰ฝ๊ณ , ๊ธˆ๋ฐฉ ๊ฒฉ์ถ”์— ๋‚˜์„คํ…Œ์ง€๋งŒ ๋ถํ•œ์ด ๋‚ด์„ธ์šฐ๋Š” ์†Œํ˜• ๋ฌด์ธ๊ธฐ๋“ค์˜ ๊ฒฝ์šฐ ์ด์•ผ๊ธฐ๊ฐ€ ๋‹ค๋ฅด๋‹ค. ์ฒ ์ƒˆ๋‚˜ ํ’์„  ๊ฐ™์€ ์ž‘์€ ๋™๋ฌผ์ด๋‚˜ ๋ฌผ์ฒด๋“ค ์กฐ์ฐจ ๋ฏธํ™•์ธ ํ•ญ์ ์œผ๋กœ ํƒ์ง€๋˜๋Š” ์ƒํ™ฉ์—์„œ ์ƒ์œ„ ๋ถ€๋Œ€๊ฐ€ ํ•ญ์  ์‹๋ณ„์— ๋‚˜์„œ๋А๋ผ ์‹œ๊ฐ„์ด ์†Œ์š”๋  ์ˆ˜ ๋ฐ–์— ์—†๋Š”๋ฐ, ๋ณธํŒ€์€ ํ•ด๋‹น ๋ฌธ์ œ๋ฅผ ํ•ด๊ฒฐํ•˜๊ธฐ ์œ„ํ•ด ๋ฌด์ธ ํ•ญ๊ณต๊ธฐ ํ•™์Šต ์ด๋ฏธ์ง€ ๋ฐ์ดํ„ฐ์…‹, ํ…Œ์ŠคํŠธ ์ด๋ฏธ์ง€ ๋ฐ์ดํ„ฐ์…‹์„ ์ˆ˜์ง‘ํ•˜์—ฌ ๋ฒ„๋“œ์ŠคํŠธ๋ผ์ดํฌ ๋ฐฉ์ง€๋ฅผ ์œ„ํ•œ ๋ชจ๋ธ์— ์ „์ดํ•™์Šต(Transfer Learning)์„ ์ถ”๊ฐ€๋กœ ์ง„ํ–‰ํ•  ์˜ˆ์ •์ด๋‹ค.

2.ํ”„๋กœ์ ํŠธ ๋ชฉํ‘œ

  • 3๊ฐœ์˜ Object Detection Model(Faster-RCNN, YOLOF, SSD) Training & Evaluation & Visualization & Inference ๋ฐ ์„ฑ๋Šฅ ์ตœ์ ํ™” ์ˆ˜ํ–‰
  • DeepSORT์— ๋Œ€ํ•œ Detection Model Training, Evaluation & ReID Model Training, Evaluation
  • ReID Model์„ ์‚ฌ์šฉํ•œ DeepSORT Model๊ณผ ReID Model์„ ์‚ฌ์šฉํ•˜์ง€ ์•Š์€ DeepSORT Model ๊ฐ„์˜ ์„ฑ๋Šฅ ๋น„๊ต

3.OpenMMLab Detection & Video Perception Toolbox and Benchmark

MMdetection: https://github.com/open-mmlab/mmdetection
MMtracking: https://github.com/open-mmlab/mmtracking

4.Model Specification

Faster-RCNN
Backbone: ResNet50
Neck: FPN
RPN_head: RPNHead
Classification Loss: CE(Cross Entropy) Loss
Bounding Box Regression Loss: L1 Loss

YOLOF
Backbone: ResNet
Neck: DilatedEncoder
BBox_head: YOLOFHead
Classification Loss: Focal Loss
Bounding Box Regression Loss: GIoU Loss

SSD
Backbone: ResNet
Neck: SSDNeck
BBox_head: SSDHead
Classification Loss: Localization Loss
Bounding Box Regression Loss: IoU Loss

5.๊ณตํ†ต SPEC & Runtime Environment

[๊ณตํ†ต Spec]
Framework: MMDetection
Learning_rate=0.02 / 8
Workers_per_gpu: 4
Batch_size: 16
Epochs: 100
Classes = ('Bird', 'Airplane', 'Helicopter', 'FighterPlane', 'Paraglidingโ€™ , 'Droneโ€™)
Visualization Tool: Tensorboard, Matplotlib

[Runtime Environment]
Sys.platform: linux
Python: 3.10.13 (main, Sep 11 2023, 13:44:35) [GCC 11.2.0]
CUDA available: True
GPU 0: NVIDIA GeForce RTX 3090
CUDA_HOME: /usr/local/cuda
NVCC: Cuda compilation tools, release 11.7, V11.7.99
GCC: gcc (Ubuntu 9.4.0-1ubuntu1~20.04.2) 9.4.0
PyTorch: 1.13.0+cu116
PyTorch compiling details: PyTorch built with:(- GCC 9.3, - C++ Version: 201402,- Intel(R) 64
architecture applications)

6.Object Detection Datasets

https://aihub.or.kr/aihubdata/data/view.do?currMenu=115&topMenu=100&aihubDataSe=realm&dataSetSn=476

AI-Hub์˜ Small object detection์„ ์œ„ํ•œ ์ด๋ฏธ์ง€ ๋ฐ์ดํ„ฐ์…‹์„ ์ด์šฉํ•˜์˜€๋‹ค. ํ•ด๋‹น ๋ฐ์ดํ„ฐ ์…‹์—์„œ๋Š” ์ด๋ฏธ์ง€(2800x2100 ํ•ด์ƒ๋„) ๋‚ด์— ์ผ์ • ํฌ๊ธฐ ์ดํ•˜์˜ ์†Œํ˜• ๊ฐ์ฒด(200x200 ํ”ฝ์…€ ํฌ๊ธฐ ์ดํ•˜)๋“ค๋งŒ ์กด์žฌํ•˜๋ฉฐ ์ด๋ฏธ์ง€์— ๋Œ€ํ•œ JSON ํ˜•ํƒœ์˜ ์–ด๋…ธํ…Œ์ด์…˜ ํŒŒ์ผ ๋˜ํ•œ ํฌํ•จํ•˜๊ณ  ์žˆ๋‹ค.

  • Type: AircraftDataset (Classes: "Bird", "Airplane", "Helicopter", "FighterPlane", "Paragliding", "Drone")
  • Train Datasets: 5760
  • Validation Datasets: 1601
  • Test Datasets: 640

7.Data Augmentation

์ถ”๊ฐ€๋กœ ๋‹ค์–‘ํ•œ ํ™˜๊ฒฝ์—์„œ์˜ ๊ฐ์ฒด ํƒ์ง€์œจ์„ ๋†’์ด๊ธฐ ์œ„ํ•ด ํ•™์Šต ๊ณผ์ • ์ค‘ ์•„๋ž˜์™€ ๊ฐ™์€ Data Augmentation ๊ธฐ๋ฒ•๋“ค์„ ์ ์šฉํ•˜์˜€๋‹ค. ํ•˜์ง€๋งŒ MMDetection์€ ํŒŒ์ดํ”„๋ผ์ธ ๋‚ด๋ถ€์—์„œ ๋ชจ๋ธ์˜ ํ•™์Šต๊ณผ ํ‰๊ฐ€๊ฐ€ ์ด๋ฃจ์–ด์ง€๊ธฐ ๋•Œ๋ฌธ์— Augmentation์ด ์ ์šฉ ๋œ ์ดํ›„์˜ ์ •ํ™•ํ•œ Datasets์˜ Size๋ฅผ ์‹๋ณ„ํ•˜๊ธฐ ๋ถˆ๊ฐ€๋Šฅํ•˜๋‹ค๋Š” Issue๊ฐ€ ์กด์žฌํ•˜์˜€๋‹ค.

  • Brightness Distortion (์ด๋ฏธ์ง€์˜ ๋ช…๋„ ๋ณ€๊ฒฝ)
    โ†’ brightness_delta=32
    โ†’ ์ตœ์†Œ๊ฐ’๊ณผ ์ตœ๋Œ€๊ฐ’์ด ๊ฐ๊ฐ -32, 32์ธ ๊ท ์ผ ๋ถ„ํฌ ํ•จ์ˆ˜์˜ Output์„ ํ†ตํ•ด ์ด๋ฏธ์ง€์˜ ๋ช…๋„๋ฅผ ๋ณ€ํ™˜ ํ•˜์˜€๋‹ค.
  • Contrast Distortion (์ด๋ฏธ์ง€์˜ ๋Œ€๋น„ ๋ณ€๊ฒฝ)
    โ†’ contrast_range=(0.5, 1.5)
    โ†’์ตœ์†Œ๊ฐ’๊ณผ ์ตœ๋Œ€๊ฐ’์ด ๊ฐ๊ฐ 0.5, 1.5์ธ ๊ท ์ผ๋ถ„ํฌํ•จ์ˆ˜์˜ Output์„ ํ†ตํ•ด ์ด์šฉํ•ด ์ด๋ฏธ์ง€์˜ ๋Œ€๋น„๋ฅผ ๋ณ€ํ™˜ ํ•˜์˜€๋‹ค. Contras Distortion์€ Brightness Distortion๊ณผ๋Š” ๋‹ค๋ฅด๊ฒŒ Sum์ด ์•„๋‹Œ Multiplication ์—ฐ์‚ฐ์„ ์ˆ˜ํ–‰ํ•˜์—ฌ Pixel Intensity์˜ ๋Œ€๋น„๋ฅผ ์ฆ๊ฐ€์‹œํ‚จ๋‹ค
  • Saturation Distortion (์ด๋ฏธ์ง€์˜ ์ฑ„๋„ ๋ณ€๊ฒฝ)
    โ†’ saturation_range=(0.5, 1.5)
    โ†’ ์ตœ์†Œ๊ฐ’๊ณผ ์ตœ๋Œ€๊ฐ’์ด ๊ฐ๊ฐ 0.5, 1.5์ธ ๊ท ์ผ ๋ถ„ํฌ ํ•จ์ˆ˜์˜ Output์„ ํ†ตํ•ด ์ด๋ฏธ์ง€์˜ ์ฑ„๋„๋ฅผ ๋ณ€ํ™˜ ํ•˜์˜€๋‹ค, H(Hue; ์ƒ‰์กฐ), S(Saturation; ์ฑ„๋„), V(Value; ๋ช…๋„)์—์„œ 1์˜ ์ธ๋ฑ์Šค๋ฅผ ๊ฐ–๋Š” S๋ฅผ ๋ณ€๊ฒฝ
  • Hue Distortion (์ด๋ฏธ์ง€์˜ ์ƒ‰์ƒ ๋ณ€๊ฒฝ)
    โ†’ hue_delta=18
  • Resize (์ด๋ฏธ์ง€์˜ ์‚ฌ์ด์ฆˆ ๋ณ€๊ฒฝ)
    โ†’ img_scale=(1333, 800)
  • RandomFlip (์ด๋ฏธ์ง€ ํšŒ์ „)
    โ†’ flip_ratio=0.5
  • Normalize (Pixel Intensity Normalization)

8.Object Detection Model ํ•™์Šต ์ˆ˜ํ–‰ ๊ฒฐ๊ณผ

[Faster-RCNN]

Faster-RCNN Learning Rate
lr
Faster-RCNN Accuracy & Train_loss
faster_rcnn_train
Faster_RCNN mAP
faster_map

[YOLOF]

YOLOF Learning Rate
yolo lr
YOLOF Train_loss
yolo tr
YOLOF mAP
yolo val

[SSD]

SSD Learning Rate
lr

SSD Train_loss
DASD

SSD mAP
bfs

9.๋ชจ๋ธ Inference ๊ฒฐ๊ณผ

asdxcz

10.Object Tracking Datasets

Detector์™€ ReID Model ํ•™์Šต ์ˆ˜ํ–‰์„ ์œ„ํ•ด ๋Œ€์ค‘ํ™”๋œ MOT datasets์˜ ๋ณดํ–‰์ž๋‚˜ ์ฐจ๋Ÿ‰ class๊ฐ€ ์•„๋‹Œ Bird class๋ฅผ ์œ„ํ•œ Custom Datasets์„ ๊ตฌ์ถ•ํ•ด์•ผ ํ–ˆ๋‹ค.๋”ฐ๋ผ์„œ CVAT Tool์„ ์ด์šฉํ•˜์—ฌ Track Rectangle๋กœ ๊ฐ๊ฐ์˜ ๊ฐ์ฒด๋ฅผ ์ง€์ •ํ•œ ๋‹ค์Œ Frame๋งˆ๋‹ค ์ƒ์ž๋ฅผ ์ด๋™์‹œ์ผœ ์ถ”์  ์ขŒํ‘œ๋ฅผ ์ €์žฅํ•˜๊ณ  ๊ฐ์ฒด๊ฐ€ ํ™”๋ฉด์—์„œ ์‚ฌ๋ผ์งˆ์‹œ Switch OFF์‹œ์ผœ Ground Truth ํŒŒ์ผ์„ ์‚ฐ์ถœํ–ˆ๋‹ค.gt.txt๋Š” ์ฐจ๋ก€๋กœ Frame number,Identity number,Bonding box left,Bounding box top,Bounding box width,Bounding box height,,Class,Visibility์ˆœ์ด๋‹ค.Train Dataset์€ Video๋ฅผ Frame Per Second ๋‹จ์œ„๋กœ ๋ถ„ํ•  ๋’ค ์ €์žฅํ•˜์˜€๋‹ค.์ตœ์ข…์œผ๋กœ MMtracking์˜ mot2coco.py,mot2reid.py ํŒŒ์ผ์„ ์ด์šฉํ•ด MOTํ˜•์‹์˜ COCO Format Annotation๊ณผ Bounding Box Image๋กœ ๊ตฌ์„ฑ๋˜์–ด ์žˆ๋Š” ReID Datasets์œผ๋กœ ๋ณ€ํ™˜ํ•˜์—ฌ Multi Object Tracking Datasets์„ ๊ตฌ์ถ•ํ•˜์˜€๋‹ค.

[CVAT tool์„ ํ™œ์šฉํ•œ Custom DataSet Labelling]

1241

[์‚ฐ์ถœ๋œ Ground Truth ํŒŒ์ผ]

asdq

[Video to Image]

vd2

[์ตœ์ข… Multi Object Tracking Dataset Structure]

zaza

11.DeepSORT๋ฅผ ์ด์šฉํ•œ Multi Object Tracking ๊ฒฐ๊ณผ

MMTracking์—์„œ ์ œ๊ณตํ•˜๋Š” DeepSORT์˜ ๊ฒฝ์šฐ Object Detection Model ๊ณผ ReID Model์„ ํ˜ผํ•ฉํ•˜์—ฌ MOT์„ ์ˆ˜ํ–‰ํ•  ์ˆ˜ ์žˆ์—ˆ๋‹ค. Detection Model์˜ ๊ฒฝ์šฐ ๋ณธ ํŒ€์ด ๊ตฌ์ถ•ํ•œ Faster-RCNN, YOLOF, SSD์˜ Checkpoint ํŒŒ์ผ์— Tracking Video์— ๋Œ€ํ•œ ์ „์ด ํ•™์Šต(Epochs: 10, Step: 10, Batch Size: 2, # of Training Datasets: 216)์„ ์ˆ˜ํ–‰ํ•œ ๋’ค ์ ์šฉํ•˜์˜€๋‹ค. ReID(Re-Identification) Model์˜ ๊ฒฝ์šฐ ์‚ฌ์šฉ๋˜๋Š” ๋ฐ์ดํ„ฐ์…‹์˜ ๊ฐ์ฒด ๊ฐ„์˜ ๊ตฌ๋ณ„๋˜๋Š” ํŠน์ง•์ด ์—†๋Š” ๊ฒฝ์šฐ, ๊ฐ์ฒด์˜ ์‹๋ณ„์ด ์–ด๋ ค์›Œ์ง€๊ณ  ์„ฑ๋Šฅ์ด ์ œํ•œ๋  ์ˆ˜ ์žˆ๋‹ค. ์ฆ‰ ๊ฐ์ฒด ๊ฐ„์˜ ์ฐจ์ด๊ฐ€ ์ถฉ๋ถ„ํžˆ ํฌ์ง€ ์•Š๊ฑฐ๋‚˜ ์œ ์˜๋ฏธํ•œ ํŠน์ง•์ด ๋ถ€์กฑํ•˜๋ฉด ๋‹ค์ค‘ ๊ฐ์ฒด์— ๋Œ€ํ•œ ์ •ํ™•ํ•œ ์‹๋ณ„๊ณผ ์ถ”์ ์ด ์–ด๋ ค์›Œ์ง€๊ณ  Generalization Performance์˜ ์ €ํ•˜๋ฅผ ์ดˆ๋ž˜ํ•  ์ˆ˜ ์žˆ๋‹ค. ๋”ฐ๋ผ์„œ ReID Model์„ ํšจ๊ณผ์ ์œผ๋กœ ํ•™์Šต์‹œํ‚ค๊ธฐ ์œ„ํ•ด์„œ๋Š” ๋ฐ์ดํ„ฐ์…‹์ด ๊ฐ์ฒด ๊ฐ„์˜ ์œ ์˜๋ฏธํ•˜๊ณ  ๊ตฌ๋ณ„๋˜๋Š” ํŠน์ง•์„ ํฌํ•จํ•˜๊ณ  ์žˆ์–ด์•ผํ•˜๋Š”๋ฐ ๋ณธ ํŒ€์˜ ํ•™์Šต ๋ฐ์ดํ„ฐ๋Š”์‚ฌ๋žŒ๊ณผ ๊ฐ™์ด ๊ตฌ๋ณ„๋˜๋Š” ํŠน์ง•์„ ๊ฐ€์ง„ ๊ฐ์ฒด๊ฐ€ ํฌํ•จ๋˜์ง€ ์•Š์•˜๊ธฐ ๋•Œ๋ฌธ์— ReID Model์„ DeepSORT์— ์ ์šฉํ•˜์˜€์„ ๋•Œ ๋ˆˆ์— ๋„๋Š” ์„ฑ๋Šฅ์˜ ํ–ฅ์ƒ์„ ๋ถˆ๋Ÿฌ์˜ฌ ์ˆ˜ ์žˆ์„์ง€ ์˜๋ฌธ์„ ๊ฐ€์ง€๊ฒŒ ๋˜์—ˆ๋‹ค. ๊ทธ๋ž˜์„œ ReID Model์„ DeepSORT์— ์ ์šฉํ–ˆ์„ ๋•Œ์™€ ์ ์šฉํ•˜์ง€ ์•Š์•˜์„ ๋•Œ์˜ ์„ฑ๋Šฅ ๋น„๊ต ์—ฐ๊ตฌ๋ฅผ ์ˆ˜ํ–‰ํ•˜์˜€๊ณ  ์•„๋ž˜์™€ ๊ฐ™์€ ๊ฒฐ๊ณผ๋ฅผ ๋„์ถœํ•  ์ˆ˜ ์žˆ์—ˆ๋‹ค.

[ReID Model Training ๊ฒฐ๊ณผ]

image

[Triplet Loss โ†’ 0.000 (figure 2)]

Triplet Loss๋Š” ์–ด๋–ค ํ•œ ๊ฐ์ฒด(Anchor)์™€ ๊ฐ™์€ ๊ฐ์ฒด(Positive), ๋‹ค๋ฅธ ๊ฐ์ฒด(Negative)์ด๋ผ๋Š” ํŒŒ๋ผ๋ฏธํ„ฐ๋ฅผ์ด์šฉํ•ด ํ•™์Šต ์‹œ ๋ฏธ๋‹ˆ ๋ฐฐ์น˜ ์•ˆ์—์„œ Anchor, Positive, Negative๋“ค์ด ์ž„๋ฒ ๋”ฉ ๋œ ๊ฐ’๋“ค์˜ ์œ ํด๋ฆฌ๋“œ ๊ฑฐ๋ฆฌ๋ฅผ ๊ตฌํ•ด ์•„๋ž˜์™€ ๊ฐ™์€ Loss ํ•จ์ˆ˜๋ฅผ ๋งŒ๋“ ๋‹ค.

image

๋Œ€๊ด„ํ˜ธ ์•ˆ์˜ ์ฒซ๋ฒˆ์งธ ํ•ญ์ด ์˜๋ฏธํ•˜๋Š” ๊ฒƒ์€ Anchor์™€ Positive๊ฐ„์˜ Distance๊ณ , ๋‘๋ฒˆ์งธ ํ•ญ์€ Anchor์™€ Negative์™€์˜ Distance์ด๋ฉฐ ฮฑ๋Š” ๋งˆ์ง„(Hyper Parameter)์„ ์˜๋ฏธํ•œ๋‹ค. ๋”ฐ๋ผ์„œ L์„ ์ตœ์†Œํ™”ํ•œ๋‹ค๋Š” ๊ฒƒ์€ Positive์™€์˜ ๊ฑฐ๋ฆฌ๋Š” ๊ฐ€๊นŒ์›Œ์ง€๋„๋ก ํ•˜๊ณ  Negative์™€์˜ ๊ฑฐ๋ฆฌ๋Š” ๋ฉ€์–ด์ง€๋„๋ก ํ•˜๋Š” ๊ฒƒ์ด๋‹ค. ์ฆ‰ Triplet Loss๊ฐ€ 0.000์ด ๋‚˜์™”๋‹ค๋Š” ๊ฒƒ์€ ReID ๋ชจ๋ธ์ด ํ•™์Šต ์ค‘์— ๋ชจ๋“  Anchor, Positive, Negative ์Œ์— ๋Œ€ํ•ด ๊ฑฐ๋ฆฌ๋ฅผ ์˜ฌ๋ฐ”๋ฅด๊ฒŒ ๊ตฌ๋ณ„ํ–ˆ๋‹ค๋Š” ์˜๋ฏธ์ด๋ฉฐ Anchor์™€ Positive ๊ฐ„์˜ ๊ฑฐ๋ฆฌ๊ฐ€ Negative ๊ฐ„์˜ ๊ฑฐ๋ฆฌ๋ณด๋‹ค ์ž‘๊ฑฐ๋‚˜ ๊ฐ™๊ฒŒ ํ•™์Šต๋˜์—ˆ๋‹ค๋Š” ๊ฒƒ์„ ๋‚˜ํƒ€๋‚ธ๋‹ค. ์ฆ‰ ์ด๋Š” ๋ชจ๋ธ์ด ์ด๋ฏธ ํ•™์Šต ๋ฐ์ดํ„ฐ์—์„œ ์ œ์‹œ๋œ ์œ ์‚ฌ์„ฑ ๋ฐ ์ฐจ์ด๋ฅผ ์ดํ•ดํ•˜๊ณ  ์žˆ์Œ์„ ์‹œ์‚ฌํ•  ์ˆ˜ ์žˆ๋‹ค.

[Top-1 Accuracy: 99.2000 (figure 4)]

Top-1 Accuracy๋ž€ Softmax Activation Function์—์„œ์˜ Output์—์„œ ์ œ์ผ ๋†’์€ ์ˆ˜์น˜๋ฅผ ๊ฐ€์ง€๋Š” ๊ฐ’์ด ์ •๋‹ต์ผ ๊ฒฝ์šฐ์— ๋Œ€ํ•œ ์ง€ํ‘œ๋ฅผ ๊ณ„์‚ฐํ•œ ๊ฒƒ์„ ์˜๋ฏธํ•œ๋‹ค. ์ฆ‰ Top-1 Accuracy๊ฐ€ 99.2000์™€ ๊ฐ™์ด ๋†’์€ ์ˆ˜์น˜๊ฐ€ ๋‚˜์™”๋‹ค๋Š” ๊ฒƒ์€ ํ•ด๋‹น ReID ๋ชจ๋ธ์ด ๋Œ€๋ถ€๋ถ„์˜ ๊ฒฝ์šฐ์— ๋Œ€ํ•ด ๊ฐ€์žฅ ๋†’์€ ํ™•๋ฅ ์„ ๊ฐ€์ง„ ํด๋ž˜์Šค๋ฅผ ์ •ํ™•ํ•˜๊ฒŒ ์‹๋ณ„ํ•œ๋‹ค๋Š” ์˜๋ฏธ์ด๋ฉฐ ์ด๋Š” ๋ชจ๋ธ์ด ์ฃผ์–ด์ง„ ๋ถ„๋ฅ˜ ์ž‘์—…์„ ์ž˜ ์ˆ˜ํ–‰ํ•˜๊ณ  ์žˆ์Œ์„ ๋‚˜ํƒ€๋‚ธ๋‹ค.

[ReID Model์„ ์‚ฌ์šฉํ•˜์ง€ ์•Š์€ ๊ฒฝ์šฐ DeepSORT ์„ฑ๋Šฅ ํ‰๊ฐ€(MOTA: 85.3%, MOTP: 0.228)]

FER

[ReID Model์„ ์‚ฌ์šฉํ•œ ๊ฒฝ์šฐ DeepSORT ์„ฑ๋Šฅ ํ‰๊ฐ€(MOTA: 93.0%, MOTP: 0.212)]

12345

๋ณธ ํŒ€์˜ ์˜ˆ์ƒ๊ณผ๋Š” ๋‹ฌ๋ฆฌ DeepSORT๋ฅผ ์ด์šฉํ•œ MOT ์ˆ˜ํ–‰์‹œ ReID Model์„ ์ด์šฉํ•œ ๊ฒฝ์šฐ MOTA(Multi Object Tracking Accuracy)์„ฑ๋Šฅ์ด 7.7% ํ–ฅ์ƒํ•˜๋ฉฐ, MOTP(Multi Object Tracking Precision) ์„ฑ๋Šฅ์€ 0.016% ํ–ฅ์ƒํ•œ ๊ฒƒ์„ ํ™•์ธํ•  ์ˆ˜ ์žˆ์—ˆ๋‹ค. ๋˜ํ•œ Recall๊ณผ Precision๊ฐ’์ด ๊ฐ๊ฐ 0.3%, 6.6% ์ฆ๊ฐ€ํ•˜์˜€๋‹ค.

12.์ตœ์ข…๊ฒฐ๊ณผ๋ฌผ ์ฃผ์š” ํŠน์ง• ๋ฐ ์„ค๋ช…

[Object Detection]

๊ณตํ†ต์œผ๋กœ ResNet๊ณ„์—ด์˜ BackBone ๊ณผ ๊ฐ๊ธฐ ๋‹ค๋ฅธ Neck,Head๋ฅผ ๊ฐ€์ง€๊ณ  ์žˆ๋‹ค.ํ•™์Šต ์ˆ˜ํ–‰ ๊ฒฐ๊ณผ์—์„œ ์ฃผ์š” ์ฐจ์ด์ ์€ 2-stage-detector Faster_RCNN์—์„  Feature Pyramid Network(FPN)์„ ์‚ฌ์šฉํ•˜์—ฌ ๋‹ค์–‘ํ•œ ์Šค์ผ€์ผ๋กœ feature map์„ ์ถ”์ถœํ–ˆ๊ณ  Cross Entropy Loss๋กœ class๋ฅผ ๋ถ„๋ฅ˜ํ•˜์—ฌ L1 Loss๋ฅผ ํ†ตํ•œ regression์œผ๋กœ ๋†’์€ accuracy๋ฅผ ๋‹ฌ์„ฑํ–ˆ๋‹ค. 1-stage-detector์ธ YOLOF๋Š” DilatedEncoder์™€ Focal Loss,GIoU Loss๋ฅผ ์‚ฌ์šฉํ–ˆ๊ณ  SSD๋Š” SSDNeck,SSDHead,Localization Loss,IoU Loss ๋ฅผ ์‚ฌ์šฉํ•˜์˜€๋‹ค. ์ข…ํ•ฉ์ ์ธ ๊ฒฐ๊ณผ: ๋ชจ๋ธ ๊ฐ๊ฐ์˜ ํŠน์ง•์— ๋งž๋Š” Neck,Head ์ ์šฉ๊ณผ Data augmentation์„ ์ ์šฉํ•˜์—ฌ 0.8์ด์ƒ์˜ ๋†’์€ mAP๋ฅผ ๋ณด์ธ๋‹ค. ํ”„๋กœ์ ํŠธ๋ฅผ ํ†ตํ•ด Small Size๋ฅผ ๊ฐ–๋Š” ์กฐ๋ฅ˜ ๋ฐ ๋น„ํ–‰๋ฌผ์ฒด๋ฅผ ๋†’์€ ์„ฑ๋Šฅ์œผ๋กœ ํƒ์ง€ํ•˜๊ณ  ์‹ค์‹œ๊ฐ„ ์ถ”์ ํ•˜๋Š” ๋ชจ๋ธ์„ ๊ตฌ์ถ•ํ•˜์˜€๋‹ค. ๋”ฐ๋ผ์„œ ํ•ด๋‹น ๋ชจ๋ธ์„ ์ ์šฉํ•œ๋‹ค๋ฉด, ์กฐ๋ฅ˜๋‚˜ ๋น„ํ–‰๋ฌผ์ฒด ๋ฟ๋งŒ ์•„๋‹ˆ๋ผ ๋‹ค๋ฅธ Small Object์— ๊ด€ํ•œ ๋†’์€ ์„ฑ๋Šฅ์˜ Detection ๋ฐ ์‹ค์‹œ๊ฐ„ ์ถ”์ ์ด ๊ฐ€๋Šฅํ•˜๋‹ค๊ณ  ๊ธฐ๋Œ€๋œ๋‹ค.

[Object Tracking]

DeepSORT์™€ ๊ด€๋ จํ•˜์—ฌ ์ด์ „์— ๋ฐœํ‘œ๋œ ๋…ผ๋ฌธ๋“ค์€ Re-identification ๋ชจ๋ธ์„ ํ†ตํ•ด ์‚ฌ๋žŒ๊ณผ ๊ฐ™์€ Object ๊ฐ„ ๊ณ ์œ ํ•˜๊ฒŒ ๊ตฌ๋ณ„๋˜๋Š” ํŠน์ง•์„ ๊ฐ–๋Š” ๋ฐ์ดํ„ฐ๋ฅผ ํ•™์Šตํ•˜๊ณ , ์ด๋กœ๋ถ€ํ„ฐ Id-switching์ด๋‚˜ Occlusion(ํ์ƒ‰) ๋ฌธ์ œ๋ฅผ ํ•ด๊ฒฐํ•˜์˜€๋‹ค. ํ•˜์ง€๋งŒ ๋ณธ ํ”„๋กœ์ ํŠธ์˜ ์‚ฌ์šฉ๋œ Training Datasets์€ Small Size์˜ ์กฐ๋ฅ˜๋‚˜ ๋น„ํ–‰๊ธฐ, ๋“œ๋ก ๊ณผ ๊ฐ™์€ ์ƒ๊ณต ๋น„ํ–‰ ๋ฌผ์ฒด์ด๊ธฐ ๋•Œ๋ฌธ์—, ์ด์ „ ๋…ผ๋ฌธ๋“ค๊ณผ ๋‹ฌ๋ฆฌ ํ•˜๋‚˜์˜ Class๋‚ด์—์„œ Object๋“ค์„ ๊ณ ์œ ํ•˜๊ฒŒ ๋ถ„๋ฅ˜ํ• ๋งŒํ•œ ํŠน์ง•์ด ์—†์„ ๊ฒƒ์ด๋ผ ์˜ˆ์ƒํ•˜์˜€๋‹ค. ๊ทธ๋Ÿฌ๋‚˜ Re-identification ๋ชจ๋ธ ํ•™์Šต ์œ ๋ฌด์— ๋”ฐ๋ผ Object Tracking ์„ฑ๋Šฅ์ด ๋‹ฌ๋ผ์ง€๋Š” ๊ฒƒ์„ ํ™•์ธํ•˜์˜€๊ณ , ์ด๋กœ๋ถ€ํ„ฐ Re-identification ๋ชจ๋ธ์ด ๋‹คํ˜•์„ฑ ๋ฐ ํ™œ์šฉ์„ฑ ๋ถ€๋ถ„์—์„œ ํ–ฅ์ƒ๋จ์„ ์ฆ๋ช…ํ•˜์˜€๋‹ค.

13.Deep Sort ๋ฐ๋ชจ ์˜์ƒ

gif_deepSORT_result

14.๊ธฐ๋Œ€ํšจ๊ณผ ๋ฐ ํ™œ์šฉ ๋ฐฉ์•ˆ

๊ฐ€. ๊ธฐ๋Œ€ํšจ๊ณผ

  • ํ•ญ๊ณต ์•ˆ์ „ ํ–ฅ์ƒโ†’ ์กฐ๋ฅ˜ ๋ฐ ๋ฌด์ธ ํ•ญ๊ณต๊ธฐ์˜ ์‹ค์‹œ๊ฐ„ ํƒ์ง€ ๋ฐ ์ถ”์ ์„ ํ†ตํ•ด ์กฐ์ข…์‚ฌ๋“ค์—๊ฒŒ ์ถฉ๋Œ ์‚ฌ๊ณ ๋ฅผ ์˜ˆ๋ฐฉํ•˜๊ณ  ํ•ญ๊ณต ์•ˆ์ „์„ฑ์„ ํ–ฅ์ƒ์‹œํ‚ฌ ๊ฒƒ์œผ๋กœ ๊ธฐ๋Œ€๋œ๋‹ค
  • ๋น„์ƒ ์ƒํ™ฉ ๋Œ€์‘ ๊ฐ•ํ™”โ†’ ๋ฌผ์ฒด ์ถ”์  ๊ธฐ์ˆ ์„ ํ™œ์šฉํ•˜์—ฌ ํ•ญ๊ณต ๋‹ด๋‹น์ž๊ฐ€ ๋น„์ƒ ์ƒํ™ฉ์— ์‹ ์†ํ•˜๊ฒŒ ๋Œ€์‘ํ•˜๊ณ  ํšจ๊ณผ์ ์œผ๋กœ ๊ด€๋ฆฌํ•  ์ˆ˜ ์žˆ๋‹ค.
  • ์ž๋™ํ™”๋œ ๋ชจ๋‹ˆํ„ฐ๋ง ์‹œ์Šคํ…œ ๋„์ž…โ†’ ํšจ์œจ์ ์ธ ๊ฐ์‹œ ๋ฐ ๊ฒฝ๋ณด ์‹œ์Šคํ…œ์„ ํ†ตํ•ด ๊ณตํ•ญ ๋ฐ ํ•ญ๊ณต ๋‹น๊ตญ์ด ์‹ค์‹œ๊ฐ„์œผ๋กœ ํ•ญ๊ณต ๊ตํ†ต ์ƒํ™ฉ์„ ๋ชจ๋‹ˆํ„ฐ๋งํ•˜๊ณ  ์œ„ํ—˜ ์ง€์—ญ์„ ์‹๋ณ„ํ•  ์ˆ˜ ์žˆ๋‹ค.
  • ๋‹ค์–‘ํ•œ ์‚ฐ์—… ๋ถ„์•ผ์—์˜ ์‘์šฉโ†’ ํ•ด์ƒ ๊ฐ์‹œ ๋ฐ ๊ฒฝ๊ณ„ ๋ณด์•ˆ, ์ž์—ฐ๋ณด์ „ ๋ฐ ํ™˜๊ฒฝ ๋ชจ๋‹ˆํ„ฐ๋ง ๋“ฑ ๋‹ค์–‘ํ•œ ๋ถ„์•ผ์— ๋ฌผ์ฒด ํƒ์ง€ ๋ฐ ์ถ”์  ๊ธฐ์ˆ ์„ ์‘์šฉํ•˜์—ฌ ํ–ฅ์ƒ๋œ ์•ˆ์ „ ๋ฐ ํšจ์œจ์„ฑ์„ ์ œ๊ณตํ•  ์ˆ˜ ์žˆ๋‹ค.
  • ์—ฐ๊ตฌ ๋ฐ ๊ต์œก ํ™œ์šฉโ†’ ์ˆ˜์ง‘๋œ ๋ฐ์ดํ„ฐ์…‹๊ณผ ๋ชจ๋ธ์€ ํ•™๊ณ„์™€ ์‚ฐ์—…๊ณ„์—์„œ ๊ฐ์ฒด ํƒ์ง€ ๋ฐ ์ถ”์ ์— ๊ด€ํ•œ ์—ฐ๊ตฌ์™€ ๊ต์œก์— ํ™œ์šฉ๋  ๊ฒƒ์œผ๋กœ ๊ธฐ๋Œ€๋œ๋‹ค.

๋‚˜. ํ™œ์šฉ๋ฐฉ์•ˆ

  • ํ•ญ๊ณต ๊ด€๋ จ ๊ธฐ๊ด€ ๋ฐ ๊ณตํ•ญ ์šด์˜โ†’ ๊ณตํ•ญ ๋ฐ ํ•ญ๊ณต ๋‹น๊ตญ์€ ๊ฐœ๋ฐœ๋œ ์‹œ์Šคํ…œ์„ ๋„์ž…ํ•˜์—ฌ ํ•ญ๊ณต ์•ˆ์ „์„ฑ์„ ํ–ฅ์ƒ์‹œํ‚ค๊ณ , ์‹ค์‹œ๊ฐ„ ๋ชจ๋‹ˆํ„ฐ๋ง์„ ํ†ตํ•ด ํ•ญ๊ณต ๊ตํ†ต์„ ํšจ์œจ์ ์œผ๋กœ ์šด์˜ํ•  ์ˆ˜ ์žˆ๋‹ค.
  • ๋น„์ƒ ์ƒํ™ฉ ๋Œ€์‘ ๋ฐ ๊ด€๋ฆฌ ๊ธฐ๊ด€โ†’ ๋น„์ƒ ์ƒํ™ฉ ๋Œ€์‘ ๊ธฐ๊ด€์€ ์‹ค์‹œ๊ฐ„ ์ถ”์  ๊ฒฐ๊ณผ๋ฅผ ํ™œ์šฉํ•˜์—ฌ ์‚ฌ๊ณ  ํ˜„์žฅ์— ๋น ๋ฅด๊ฒŒ ๋Œ€์‘ํ•˜๊ณ , ์ƒํ™ฉ์„ ํšจ๊ณผ์ ์œผ๋กœ ๊ด€๋ฆฌํ•  ์ˆ˜ ์žˆ๋‹ค.
  • ํ•ด์ƒ ๊ฐ์‹œ ๋ฐ ๊ฒฝ๊ณ„ ๋ณด์•ˆโ†’ ํ•ด๊ตฐ ๋ฐ ๊ฒฝ๋น„ ๋‹น๊ตญ์€ ๋ฌผ์ฒด ํƒ์ง€ ๋ฐ ์ถ”์  ๊ธฐ์ˆ ์„ ํ•ด์ƒ ๊ฐ์‹œ์— ํ™œ์šฉํ•˜์—ฌ ์นจ์ž…์ด๋‚˜ ์œ„ํ—˜ํ•œ ์ƒํ™ฉ์„ ์‹ ์†ํžˆ ๊ฐ์ง€ํ•˜๊ณ  ๋Œ€์‘ํ•  ์ˆ˜ ์žˆ๋‹ค.
  • ํ™˜๊ฒฝ ๋ณด์ „ ๋ฐ ์ƒํƒœํ•™ ์—ฐ๊ตฌโ†’ ํ™˜๊ฒฝ ๋‹น๊ตญ ๋ฐ ์—ฐ๊ตฌ ๊ธฐ๊ด€์€ ๋ฌผ์ฒด ์ถ”์  ๊ธฐ์ˆ ์„ ์ด์šฉํ•˜์—ฌ ์กฐ๋ฅ˜ ๋ฐ ๋™๋ฌผ์˜ ์ด๋™์„ ์—ฐ๊ตฌํ•˜๊ณ  ํ™˜๊ฒฝ ๋ณด์ „ ํ™œ๋™์— ํ™œ์šฉํ•  ์ˆ˜ ์žˆ๋‹ค.
  • ์—ฐ๊ตฌ ๋ฐ ๊ต์œก ๊ธฐ๊ด€โ†’ ๋Œ€ํ•™ ๋ฐ ์—ฐ๊ตฌ ๊ธฐ๊ด€์€ ์ˆ˜์ง‘๋œ ๋ฐ์ดํ„ฐ์…‹๊ณผ ๋ชจ๋ธ์„ ํ™œ์šฉํ•˜์—ฌ ๊ฐ์ฒด ํƒ์ง€ ๋ฐ ์ถ”์ ์— ๊ด€ํ•œ ์—ฐ๊ตฌ๋ฅผ ์ˆ˜ํ–‰ํ•˜๊ณ , ๊ต์œก ๊ณผ์ •์— ํ™œ์šฉํ•  ์ˆ˜ ์žˆ๋‹ค.

15.๊ฒฐ๋ก  ๋ฐ ์ œ์–ธ

ํ”„๋กœ์ ํŠธ ๊ฒฐ๊ณผ๋กœ ์–ป์€ ๋ชจ๋ธ๊ณผ ๋ฐ์ดํ„ฐ๋Š” ํ•ญ๊ณต ๋ฐ ๊ตญ๋ฐฉ ๊ด€๋ จ ์‚ฐ์—…์—์„œ์˜ ์•ˆ์ „ ๋ฐ ํšจ์œจ์„ฑ ํ–ฅ์ƒ์— ํฐ ๊ธฐ์—ฌ๋ฅผ ํ•  ๊ฒƒ์œผ๋กœ ๊ธฐ๋Œ€๋œ๋‹ค. ๋ฏธ๋ž˜์—๋Š” ๋” ๋งŽ์€ ๋ฐ์ดํ„ฐ๋ฅผ ์ˆ˜์ง‘ํ•˜๊ณ  ๋ชจ๋ธ์„ ํŠœ๋‹ํ•˜์—ฌ ๋‹ค์–‘ํ•œ ์ƒํ™ฉ์—์„œ์˜ ์ ์šฉ ๊ฐ€๋Šฅ์„ฑ์„ ๋†’์ผ์ˆ˜ ์žˆ์„๊ฒƒ์ด๋‹ค.

Popular repositories Loading

  1. SSD_KIM SSD_KIM Public

    Python

  2. YOLOF_LEE YOLOF_LEE Public

    Jupyter Notebook

  3. Koo Koo Public

    Detection and tracking of abnormal objects in the air in various weather environments

    Jupyter Notebook 1

  4. .github .github Public

Repositories

Showing 4 of 4 repositories

Top languages

Loadingโ€ฆ

Most used topics

Loadingโ€ฆ