search

WoodsGao / rotatable_yolov3

A rotatable yolov3 model which can regress the angle of the bounding box
GNU General Public License v3.0
19 stars 5 forks source link

兄弟，斜矩形的json文件是怎么生成的，有工具吗？json内容是什么，只找到正矩形的json内容 #4

Open 2387649299 opened 4 years ago

WoodsGao commented 4 years ago

标准使用coco格式的polygon，dataset类自动实现了对polygon求最接近的旋转框的过程，具体可以参考datasets的实现方式

2387649299 commented 4 years ago

ImportError: cannot import name 'mobilenet_v2' from 'pytorch_modules.backbones'

yolov3.py中库文件mobilenet_v2、pytorch_modules.backbones.mobilenet、replace_relu6、我在pytorch-modules0.3.5中没有找到对应文件，楼主用的是哪个版本的，是我装错版本了吗？

（json文件已找到一个相关工具，转换成功了）

from pytorch_modules.backbones import (mobilenet_v2, resnet34, resnet50,resnext50_32x4d) from pytorch_modules.backbones.mobilenet import ConvBNReLU, InvertedResidual from pytorch_modules.utils import initialize_weights, replace_relu6

WoodsGao commented 4 years ago

直接进入pytorch_modules的repo中手动安装最新版

2387649299 commented 4 years ago

大佬，我使用的是MSRA-TD500数据集，执行了以下操作 1.将txt转成json后用你说的工具转成train.json,val.json。(转json代码如下) convert_to_coco.zip 2.将所有图片和两个json文件放到/home/twd/github/rotatable_yolov3/data/目录下。 3.运行python train.py data/ 结果报错：RuntimeError: CUDA error: device-side assert triggered

希望大佬能添加一下简单的训练数据集（十几张图片和对应的json文件）或帮忙指出一下我操作中的问题。 （2387649299就是本人qq，如果能qq询问就更好了，嘿嘿） 我个人能力太菜了，试了GitHub好几个yolo rotated-detect都跑不起来

WoodsGao commented 4 years ago

请检查一下num_class相关的参数。 有全职工作所以暂不提供技术支持，代码更新欢迎提交pull request。

Jingzhuo-Liang commented 3 years ago

ImportError: cannot import name 'mobilenet_v2' from 'pytorch_modules.backbones'

yolov3.py中库文件mobilenet_v2、pytorch_modules.backbones.mobilenet、replace_relu6、我在pytorch-modules0.3.5中没有找到对应文件，楼主用的是哪个版本的，是我装错版本了吗？

（json文件已找到一个相关工具，转换成功了）

from pytorch_modules.backbones import (mobilenet_v2, resnet34, resnet50,resnext50_32x4d) from pytorch_modules.backbones.mobilenet import ConvBNReLU, InvertedResidual from pytorch_modules.utils import initialize_weights, replace_relu6

兄弟，你遇到的这个问题是怎么解决的啊，我也遇到了