yolov5_obb

yolov5_obb copied to clipboard

cannot import name 'nms_rotated_ext'

1

I am trying to run a demo program to train the original dataset demo, but getting ImportError: cannot import name 'nms_rotated_ext' from partially initialized module 'utils.nms_rotated' (most likely due to...

chyphen7

Custom Dataset Training

8

     

hulinlang

bug when trying to save txt bboxes with detect.py --save-txt

5

In Colab ``` %cd /content/yolov5_obb !python detect.py --weights '/content/best.pt' --source '../datasets/sample/valid/images' --conf-thres 0.5 --img 640 --device 0 --agnostic-nms --save-txt ``` Result ``` /content/yolov5_obb detect: weights=['/content/sample_data/best.pt'], source=../datasets/sample/valid/images, imgsz=[640, 640], conf_thres=0.5, iou_thres=0.4,...

vshesh

[Torch1.11 error] src/poly_nms_cuda.cu:4:10: fatal error: THC/THC.h: No such file or directory

8

之前的环境是ubuntu18.04，python3.7，cuda10.2 现在上机载电脑，换了环境，ubuntu20.04，python3.8，cuda11.4，torch1.11，运行detect.py出现导包报错： ``` nvidia@ubuntu:~/ros_workspace/src/yolov5_ros/src/yolov5_obb-master/yolov5_obb-master$ python3 detect.py Traceback (most recent call last): File "detect.py", line 32, in from models.common import DetectMultiBackend File "/home/nvidia/ros_workspace/src/yolov5_ros/src/yolov5_obb-master/yolov5_obb-master/models/common.py", line 23, in from utils.datasets import exif_transpose, letterbox...

tinly00

Custom Dataset Training

1

 

wanghui589

koshinryuu

val.py: cannot import name 'nms_rotated_ext' from 'utils.nms_rotated'

6

val: WARNING: dataset/val/images/0200__1__0___0.jpg: ignoring corrupt image/label: 'dict' object has no attribute 'index' val: New cache created: dataset/val/labelTxt.cache Traceback (most recent call last): File "/home/liujiafei/yolov5_obb/val.py", line 401, in main(opt) File "/home/liujiafei/yolov5_obb/val.py",...

liujiafei666

[BUG in torch1.11]THCDeviceUtils.cuh在pytorch1.11版本已经弃用，编译nms_rotated 失败 (yolov5_obb/utils/nms_rotated/src/poly_nms_cuda.cu:5:10: fatal error: THC/THCDeviceUtils.cuh: No such file or directory)

15

参考 https://github.com/pytorch/pytorch/pull/65472

QingYuan-L

support pytorch1.11

12

according to https://github.com/pytorch/pytorch/pull/65492 and https://github.com/pytorch/pytorch/pull/65472

QingYuan-L

Run yolov5_obb on a Smartphone ?

14

Hi, I was wondering if it might be possible to run your model even on the limited computing power of a smartphone. Before diving too deep into it, I wanted...

Avant-Gardiste