Commit 05e72829 authored by bailuo's avatar bailuo
Browse files

update readme

parent eb977402
...@@ -4,10 +4,11 @@ ...@@ -4,10 +4,11 @@
## 论文 ## 论文
`Tracking Everything Everywhere All at Once` `Tracking Everything Everywhere All at Once`
- https://arxiv.org/abs/2306.05422 - https://arxiv.org/abs/2306.05422
- ICCV 2023, Oral, Best Student Paper
## 模型结构 ## 模型结构
<!-- 此处一句话简要介绍模型结构 --> <!-- 此处一句话简要介绍模型结构 -->
先把一个序列表示成一个准3D的规范量,然后通过定义一个双射,这样我们通过一个准3D空间,就可以描述一个完整的运动。 先把一个序列表示成一个准3D的规范量,然后通过定义一个双射,这样就可以通过一个准3D空间,描述一个完整的运动。
<div align=center> <div align=center>
<img src="./doc/Method_overview.png"/> <img src="./doc/Method_overview.png"/>
</div> </div>
...@@ -25,7 +26,7 @@ mv omnimotion_pytoch omnimotion # 去框架名后缀 ...@@ -25,7 +26,7 @@ mv omnimotion_pytoch omnimotion # 去框架名后缀
# -v 路径、docker_name和imageID根据实际情况修改 # -v 路径、docker_name和imageID根据实际情况修改
``` ```
### Docker(方法一) ### Docker(方法一)
此处提供[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤 <!-- 此处提供[光源](https://www.sourcefind.cn/#/service-details)拉取docker镜像的地址与使用步骤 -->
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-ubuntu20.04-dtk23.10-py38 # 本镜像imageID为:0a56ef1842a7 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-ubuntu20.04-dtk23.10-py38 # 本镜像imageID为:0a56ef1842a7
docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=16G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=16G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
...@@ -34,7 +35,7 @@ pip install -r requirements.txt ...@@ -34,7 +35,7 @@ pip install -r requirements.txt
``` ```
### Dockerfile(方法二) ### Dockerfile(方法二)
此处提供dockerfile的使用方法 <!-- 此处提供dockerfile的使用方法 -->
``` ```
cd /your_code_path/omnimotion/docker cd /your_code_path/omnimotion/docker
docker build --no-cache -t codestral:latest . docker build --no-cache -t codestral:latest .
...@@ -64,7 +65,7 @@ pip install -r requirements.txt ...@@ -64,7 +65,7 @@ pip install -r requirements.txt
<!-- - 此处填写公开数据集在公司内部的下载地址(数据集存放中心为:[SCNet AIDatasets](http://113.200.138.88:18080/aidatasets) ,模型用到的各公开数据集请分别填上具体地址。),过小权重文件可打包到项目里。 --> <!-- - 此处填写公开数据集在公司内部的下载地址(数据集存放中心为:[SCNet AIDatasets](http://113.200.138.88:18080/aidatasets) ,模型用到的各公开数据集请分别填上具体地址。),过小权重文件可打包到项目里。 -->
<!-- - 此处填写公开数据集官网下载地址(非必须)。 --> <!-- - 此处填写公开数据集官网下载地址(非必须)。 -->
提供数据下载、预处理脚本的使用方法 数据下载、预处理脚本的使用方法
``` ```
cd /your_code_path/omnimotion/ cd /your_code_path/omnimotion/
python get_davis.py # 下载数据集DAVIS-2017-trainval-480p python get_davis.py # 下载数据集DAVIS-2017-trainval-480p
...@@ -84,17 +85,19 @@ python main_processing.py # 预处理数据集 ...@@ -84,17 +85,19 @@ python main_processing.py # 预处理数据集
├──sequence_2/ ├──sequence_2/
├──... ├──...
``` ```
## 训练 ## 训练
<!-- 一般情况下,ModelZoo上的项目提供单机训练的启动方法即可,单机多卡、单机单卡至少提供其一训练方法。 --> <!-- 一般情况下,ModelZoo上的项目提供单机训练的启动方法即可,单机多卡、单机单卡至少提供其一训练方法。 -->
### 单机多卡 <!-- ### 单机多卡
``` ```
python train.py --config configs/default.txt # 注意修改configs文件以及config.py文件 python train.py --config configs/default.txt # 注意修改configs文件以及config.py文件
``` ``` -->
<!-- ### 单机单卡 ### 单机单卡
``` ```
sh xxx.sh 或python xxx.py python train.py --config configs/default.txt # 注意修改configs文件以及config.py文件
``` --> ```
## 推理 ## 推理
``` ```
python viz.py --config configs/default.txt python viz.py --config configs/default.txt
...@@ -149,11 +152,5 @@ python viz.py --config configs/default.txt ...@@ -149,11 +152,5 @@ python viz.py --config configs/default.txt
- https://developer.hpccube.com/codes/modelzoo/omnimotion_pytorch - https://developer.hpccube.com/codes/modelzoo/omnimotion_pytorch
## 参考资料 ## 参考资料
<!-- - 此处填源github地址(方便使用者查看原github issue)
- 此处填参考项目或教程网址 -->
- https://github.com/qianqianwang68/omnimotion - https://github.com/qianqianwang68/omnimotion
<!-- `关于model.properties(必要)、LICENSE(必要)、CONTRIBUTORS、模型图标(必要)等其它信息提供参照: `[`ModelZooStd.md`](./ModelZooStd.md)
`各个模型需要保留原项目README.md,改名为README_origin.md即可。` -->
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment