Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
OmniMotion_pytorch
Commits
05e72829
Commit
05e72829
authored
Jul 16, 2024
by
bailuo
Browse files
update readme
parent
eb977402
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
12 additions
and
15 deletions
+12
-15
README.md
README.md
+12
-15
README_origin.md
README_origin.md
+0
-0
No files found.
README.md
View file @
05e72829
...
@@ -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即可。`
-->
README_o.md
→
README_o
rigin
.md
View file @
05e72829
File moved
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment