深度学习
文章平均质量分 66
dragonfly91
这个作者很懒,什么都没留下…
展开
-
Kafka连接报错:kafka.errors.NoBrokersAvailable: NoBrokersAvailable
最近做的项目,搭建的kafka服务器要通过外网进行生产和消费,但是发现不管是生产还是消费都会卡住,消息发不出去,但是topic可以正常查看,kafka服务也能初始化。最后发现需要修改server.properties中的一行,然后重启kafka和zookeeper服务即可。修改如下#listeners=PLAINTEXT://:9092advertised.listeners=PLAINTEXT://x.x.x.x:9092x.x.x.x代表服务器对外的IP...原创 2021-11-18 09:56:34 · 4598 阅读 · 1 评论 -
pth->onnx->trt踩坑笔记1
1、最近编译一个torch的pth转TensorRT的trt工具的时候遇到的错误:/usr/bin/ld: 找不到 -lnvinfer/usr/bin/ld: 找不到 -lcudartcollect2: error: ld returned 1 exit statusmake[2]: *** [CMakeFiles/dbnet.dir/build.make:120:dbnet] 错误 1make[1]: *** [CMakeFiles/Makefile2:95:CMakeFiles/db原创 2021-01-26 17:31:30 · 1919 阅读 · 3 评论 -
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 · 9970 阅读 · 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 评论 -
U-Net论文详解
U-Net:生物医学图像分割的卷积神经网络U-net 是基于FCN的一个语义分割网络,适合用来做医学图像的分割。摘要有许多成功利用大量带标注训练数据的神经网络。在这篇论文里,我们提出一个网络和训练策略,更有效的利用了数据增强来使用可获得的带标注数据。这个算法包括一个收缩路径来捕捉语义和一个扩展路径来精准定位。这个网络可以端到端训练非常少的图片,但是在ISBI(电子显微镜的细胞图像分...转载 2019-04-21 20:50:02 · 706 阅读 · 0 评论