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

update readme

parent eb977402
......@@ -4,10 +4,11 @@
## 论文
`Tracking Everything Everywhere All at Once`
- https://arxiv.org/abs/2306.05422
- ICCV 2023, Oral, Best Student Paper
## 模型结构
<!-- 此处一句话简要介绍模型结构 -->
先把一个序列表示成一个准3D的规范量,然后通过定义一个双射,这样我们通过一个准3D空间,就可以描述一个完整的运动。
先把一个序列表示成一个准3D的规范量,然后通过定义一个双射,这样就可以通过一个准3D空间,描述一个完整的运动。
<div align=center>
<img src="./doc/Method_overview.png"/>
</div>
......@@ -25,7 +26,7 @@ mv omnimotion_pytoch omnimotion # 去框架名后缀
# -v 路径、docker_name和imageID根据实际情况修改
```
### 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 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
```
### Dockerfile(方法二)
此处提供dockerfile的使用方法
<!-- 此处提供dockerfile的使用方法 -->
```
cd /your_code_path/omnimotion/docker
docker build --no-cache -t codestral:latest .
......@@ -64,7 +65,7 @@ pip install -r requirements.txt
<!-- - 此处填写公开数据集在公司内部的下载地址(数据集存放中心为:[SCNet AIDatasets](http://113.200.138.88:18080/aidatasets) ,模型用到的各公开数据集请分别填上具体地址。),过小权重文件可打包到项目里。 -->
<!-- - 此处填写公开数据集官网下载地址(非必须)。 -->
提供数据下载、预处理脚本的使用方法
数据下载、预处理脚本的使用方法
```
cd /your_code_path/omnimotion/
python get_davis.py # 下载数据集DAVIS-2017-trainval-480p
......@@ -84,17 +85,19 @@ python main_processing.py # 预处理数据集
├──sequence_2/
├──...
```
## 训练
<!-- 一般情况下,ModelZoo上的项目提供单机训练的启动方法即可,单机多卡、单机单卡至少提供其一训练方法。 -->
### 单机多卡
<!-- ### 单机多卡
```
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
......@@ -149,11 +152,5 @@ python viz.py --config configs/default.txt
- https://developer.hpccube.com/codes/modelzoo/omnimotion_pytorch
## 参考资料
<!-- - 此处填源github地址(方便使用者查看原github issue)
- 此处填参考项目或教程网址 -->
- 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