计算机视觉
dragonfly91
这个作者很懒,什么都没留下…
展开
-
Pascal Voc格式转ICDAR2015格式的脚本
最近做OCR的项目,需要将Pascal Voc的格式转换为ICDAR2015的格式,所以写了这个转换脚本。importosimportxml.etree.ElementTreeasETxmldir='Annotations'gtdir='ICDAR_gt'ifnotos.path.exists(gtdir):os.mkdir(gtdir)xmlfiles=os.listdir(xmldir)forxmlfileinxmlfiles:...原创 2020-11-23 10:52:40 · 396 阅读 · 1 评论 -
Pytorch模型转换,pth->onnx->trt(TensorRT engine)
pth->onnx常见问题##模型输入输出不支持字典在使用torch.onnx导出onnx格式时,模型的输入和输出都不支持字典类型的数据结构。**解决方法:**此时,可以将字典数据结构换为torch.onnx支持的列表或者元组。例如:heads {'hm': 1, 'wh': 2, 'hps': 34, 'reg': 2, 'hm_hp': 17, 'hp_offset'...原创 2019-11-28 18:41:48 · 9971 阅读 · 4 评论 -
DCNv2编译找不到nvcc问题
在编译DCNv2 0.1版本的时候,遇到nvcc库找不到的问题,因为系统默认的nvcc库路径是/usr/bin/nvcc,此时需要手动设置相关的LD_LIBRARY_PATH。我的.bashrc的修改如下:export PATH=/usr/local/cuda/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_...原创 2019-11-21 12:24:55 · 1523 阅读 · 1 评论