Skip to content

Jiahao-Ma/2D-3D-IoUs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

8033c1b · Sep 7, 2021

History

3 Commits
Sep 7, 2021
Sep 7, 2021
Sep 7, 2021
Sep 7, 2021
Sep 7, 2021

Repository files navigation

Introduction

This repo is an unofficial implementation of IoU Loss for 2D/3D Object Detection. It contains the simple calculattion of IoUs of 2D / 3D rotated bounding box.

Requirements

Following dependencies are needed

cudatoolkit=10.2
pytorch>1.5         
numpy
matplotlib

Usage

First, compile the CUDA extension.

cd cuda_op
python setup.py install

Then, run a demo which validate the Pytorch functions and CUDA extension.

cd ..
python IoUDemo.py

Future Work

Complete the implementation of GIoU-loss and DIoU-loss for rotated bounding boxes.

About

IoU of 2D / 3D rotated bounding box by Pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published