search

XinzeLee / RotateObjectDetection

This repository is based on Ultralytics/yolov5, with adjustments to enable rotate prediction boxes.
113 stars 22 forks source link

Inference not working #23

Closed satpalsr closed 2 years ago

satpalsr commented 2 years ago

Hey @XinzeLee, Tutorial-1 colab does not detect bounding boxes.

Steps to reproduce:

  1. !git clone https://github.com/XinzeLee/RotateObjectDetection
  2. %cd /content/RotateObjectDetection/rotate-yolov5
  3. !pip install -r requirements.txt
  4. %cd /content/RotateObjectDetection/rotate-yolov5/utils/iou_cuda
  5. !python setup.py install
  6. %cd /content/RotateObjectDetection/rotate-yolov5
  7. Below code 
    
from IPython.display import Image

!python rotate_detect.py --weights rotate-yolov5s-ucas.pt --img 1024 --conf 0.75 \ --source data/images/UCAS-AOD --iou-thres 0.4 --hide-labels

Image(filename='runs/detect/exp/1070.png', width=1024)


8. Check out images under runs/detect/exp
satpalsr commented 2 years ago

Train model for higher epochs instead of using the pre-trained available model.