深度学习&计算机视觉
文章平均质量分 55
RAFA CHANG
这个作者很懒,什么都没留下…
展开
-
faster-rcnn数据处理总结
reldb->blob图像增强水平翻转step 1:修改reldb具体来说,reldb本身是由5个属性组成的,所以生成的新reldb还是要按照这个格式,区别在最后 1. def get_training_roidb(imdb) 2. imdb.append_flipped_images_rel() 因为这个函数是imdb特有的,所以函数属于imdb.py imdb.py: 3. widths = self._get_widths_rel() 获得按照顺序的图片宽度 vg.py (因原创 2021-08-18 23:46:34 · 272 阅读 · 0 评论 -
将pytorch的DataParallel转移到DistributedDataParallel
需要转移的原因DistributedDataParallel是多进程的,并且支持多机并行训练。且哪怕只有1个gpu,也会提升20%的速度几个概念目标是一个进程控制一个gpuargs.world_size = args.gpus * args.nodes #每个节点内的GPU个数×节点数def setup(rank, world_size): os.environ['MASTER_ADDR'] = 'localhost' os.environ['MASTER_PORT'] = '1原创 2021-08-18 22:06:42 · 300 阅读 · 0 评论 -
pytorch编写cuda/c++ extention 方法
起因我在尝试一篇目标识别的论文的开源代码时,发现由于其NMS、LSTM等模块是使用c++编译的(为了弥补python速度的短板),对Pytorch要求不能使用超过0.3的版本。和FASTER RCNN开源代码很像,不过Faster-rcnn官方更新了其包含的c++文件,从而可以在pytorch大于1的版本上使用。我不想退回pytorch 0.3版本,于是尝试重写c++/cuda 扩展的模块prerequisiteCUDA10.0pytorch1.4+cu100torchvision0.5.0+cu原创 2021-01-31 23:56:18 · 1563 阅读 · 1 评论 -
second.pytorch、pointpillars运行踩坑大全
欢迎使用Markdown编辑器numony的pytorch.second主要是做pointpillars的,下载kitti后要建立imageset文件夹,并使用脚本来将编号拆分为train test eval三部分。否则无法运行create data.py要想使用另外一块gpu,改变train.py里面的cuda:0是会报错的,正确方法是在python3 train.py前面加上CUDA_VISIBLE_DEVICES=1使用traveller58的second.pytorch是为了使用可视化,但是原创 2020-07-09 21:01:56 · 1355 阅读 · 0 评论 -
在ros里使用Gige相机,并转换图像格式为opencv mat
最近在做无人驾驶方面的东西,老师给了一个摄像头,是Gige协议,公司叫microview的不知名公司,连驱动也只有ubuntu16.04的。百度一搜,有关ROS的都是USB摄像头,仔细想想,USB摄像头还是小儿科了一点,性能应该不如千兆网口的。基本现在市面买的摄像头也没提供ros方案。于是只能自己写package。用千兆网口摄像机的几种解决方案1千兆网口,但协议是rstp之类的网络摄像头。这种摄像头是输出h.265或者MJPEG之类的格式。不管在opencv还是ros都容易调用2 gige/gig v原创 2020-06-25 15:31:52 · 1624 阅读 · 1 评论 -
could not build wheels for scipy which use PEP 517 and cannot be installed directly解决
前几天,给我的jetson xavier重新刷系统,刷的是jetpack4.3版本。之后在pip3 install keras的时候,始终卡在安装scipy的地方,经过一番努力,最后报错内容大致是“could not build wheels for scipy which use PEP 517 and cannot be installed directly”这个问题困扰到我打算重新刷机,最...原创 2020-04-05 17:27:09 · 12325 阅读 · 5 评论 -
服务器配置深度学习环境
一共有两种方法首先先执行sudo apt-get install libc6-dev build-essential亲测如果不执行,手动安装nvidia driver会报错使用apt的方法(比较方便,简单)参考了tensoeflow官网的安装方法,Ubuntu 18.04 (CUDA 10.1) # Add NVIDIA package repositories wget...原创 2020-03-20 20:00:07 · 401 阅读 · 0 评论