SA-BEV复现过程(环境配置+Docker快速入门记录)

最近在复现论文过程中使用了docker,感觉很方便,记录一下防止自己忘了,只做记录用

SA-BEV: Generating Semantic-Aware Bird's-Eye-View Feature for Multi-view 3D Object Detection

本文的大致思路就是基于bevdepth引入了语义信息的监督,提高深度估计对于前景物体的准确性,具体可以看论文,几个模块设计还是挺有意思的,这里就不做介绍了

过程

github上面提供了dockerfile

由于感觉配conda环境繁琐,既然提供了docker就直接配docker吧,先build一个镜像

cd docker && docker build . -t [name]

name就是你的镜像的名字

在得到镜像之后,我们可以基于这个镜像创建container,在container下面运行我们的代码。

为了创建container,需要建立一个.sh文件,

.sh文件如下所示

xhost +local:docker
# sleep 1
# redis-cli <<EOF
# SELECT 4
# FLUSHDB
# EOF
sleep 1
docker stop [container's name] 
docker rm [container's name] 
docker run \
--gpus all \
--name=[container's name] \
--net=host \
-itd \
--shm-size="32g" \
-v /tmp/.X11-unix:/tmp/.X11-unix \
-v /cv/:/cv/ \
-v /home/westwell/Downloads:/home/westwell/Downloads \
-v /home/westwell/Documents:/home/westwell/Documents \
-v /etc/timezone:/etc/timezone \
-v /etc/localtime:/etc/localtime \
-e DISPLAY=unix$DISPLAY -e GDK_SCALE -e GDK_DPI_SCALE \
-e NVIDIA_DRIVER_CAPABILITIES=compute,utility,video \
[image_name]

值得注意的是,我们这段脚本是为了在container里进行一些可视化操作,例如open3d可视化点云时不会跳出display相关的错误。(网上的其他教程也是对的,但是只是能保证运行)

就ok了

一些注意点

作者的dokerfile中有一些tensorRT的内容,(和落地相关?),可能会导致build不成功,因此我注释掉了,再build就成功了。也能够正常运行:

对于docker的一些个人理解

dockerfile相当于开发者对于自己的环境做了个文本表达,你可以用这个文件创建一个他的开发环境的image(镜像),然后为了能够运行这个镜像,需要基于镜像开启一个container(容器),一个image可以对应多个container(不能重名!),然后你在container里运行代码就ok了,当然也可以在里面装conda扩展之类的,新增环境等等。。。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值