https://github.com/ksachdeva/od-loss-exp

Some work to compare various object detection loss functions
https://github.com/ksachdeva/od-loss-exp

Last synced: 11 months ago
JSON representation

Some work to compare various object detection loss functions

• Host: GitHub
• URL: https://github.com/ksachdeva/od-loss-exp
• Owner: ksachdeva
• License: apache-2.0
• Created: 2023-07-10T13:59:06.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2023-07-11T18:48:22.000Z (over 2 years ago)
• Last Synced: 2025-03-24T06:04:46.637Z (11 months ago)
• Language: Jupyter Notebook
• Size: 30.7 MB
• Stars: 4
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# Object Detection Loss comparison

Some work to compare various object detection loss functions

## Confusion Matrix

GIou             |  DIoU |                                  CIoU

:-------------------------:|:-------------------------:|:-------------------------:

![](yv5-n-giou/confusion_matrix.png)  |  ![](yv5-n-diou/confusion_matrix.png) | ![](yv5-n-ciou/confusion_matrix.png)

## mAP50-95 (on VOC test dataset)

GIou             |  DIoU |                                  CIoU

:-------------------------:|:-------------------------:|:-------------------------:

0.458  |  0.461 | 0.465

## Validation Summary (using best.pt)

GIou             |  DIoU |                                  CIoU

:-------------------------:|:-------------------------:|:-------------------------:

![](yv5-n-giou/val_summary.png)  |  ![](yv5-n-diou/val_summary.png) | ![](yv5-n-ciou/val_summary.png)

## Training time (in hours)

GPU - 1080Ti

GIou             |  DIoU |                                  CIoU

:-------------------------:|:-------------------------:|:-------------------------:

13.802   |  13.639 | 14.384