/Use_UnrealCV

Object detection using unrealcv

Primary LanguagePythonApache License 2.0Apache-2.0

Object detection using UnrealCV

UnrealCV 가상 시뮬레이션 환경에서 detectron2를 이용한 object detection을 진행한다.

detectron2는 Facebook AI Research에서 제공하는 객체 감지 알고리즘으로, pytorch deep learning framework에 의해 구동된다.

환경 (요구 사항)

  • Ubuntu 16.04
  • python 3.8 (3.6 이상)
  • GTX 1060
  • PyTorch 1.7.1 (1.5 이상)
  • CUDA 10.1
  • OpenCV 4.5.1

1. Repository cloning

git clone https://github.com/boookk/Use_UnrealCV.git

2. Create virtual environment

cd Use_UnrealCV

conda env create -f environment.yml

conda activate urdr

3. Installing UnrealCV

다양한 시뮬레이션 환경 중 'UnrealCV'를 사용하였다.

UnrealCV는 Unreal Engine 4 (UE4)를 사용한 시뮬레이션으로 Model Zoo에서 컴파일된 게임 바이너리를 제공한다.

Model Zoo에서 제공하는 게임 바이너리를 다운 받는다. (Ex. RealisticRendering)

sh virtual_env.sh

위의 명령어를 입력하거나, Model Zoo 링크를 통해 게임 바이너리를 다운 받을 수 있다.

4. Run

Object detection

  • Instance Segmentation (+ bbox, accuracy)
python code/instance_Seg.py
  • Semantic Segmentation
python code/semantic_Seg.py