Browse Source

[to #42322933]fix file header in video_single_object_tracking

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10195146
master
lanjinpeng.ljp yingda.chen 3 years ago
parent
commit
7f2a781a47
12 changed files with 23 additions and 11 deletions
  1. +2
    -1
      modelscope/models/cv/video_single_object_tracking/config/ostrack.py
  2. +2
    -1
      modelscope/models/cv/video_single_object_tracking/models/layers/attn.py
  3. +2
    -1
      modelscope/models/cv/video_single_object_tracking/models/layers/attn_blocks.py
  4. +2
    -1
      modelscope/models/cv/video_single_object_tracking/models/layers/head.py
  5. +2
    -1
      modelscope/models/cv/video_single_object_tracking/models/layers/patch_embed.py
  6. +2
    -1
      modelscope/models/cv/video_single_object_tracking/models/ostrack/base_backbone.py
  7. +2
    -1
      modelscope/models/cv/video_single_object_tracking/models/ostrack/ostrack.py
  8. +2
    -1
      modelscope/models/cv/video_single_object_tracking/models/ostrack/utils.py
  9. +2
    -1
      modelscope/models/cv/video_single_object_tracking/models/ostrack/vit_ce.py
  10. +2
    -1
      modelscope/models/cv/video_single_object_tracking/tracker/ostrack.py
  11. +2
    -1
      modelscope/models/cv/video_single_object_tracking/utils/utils.py
  12. +1
    -0
      modelscope/pipelines/cv/video_single_object_tracking_pipeline.py

+ 2
- 1
modelscope/models/cv/video_single_object_tracking/config/ostrack.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
from easydict import EasyDict as edict

cfg = edict()


+ 2
- 1
modelscope/models/cv/video_single_object_tracking/models/layers/attn.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
import torch.nn as nn




+ 2
- 1
modelscope/models/cv/video_single_object_tracking/models/layers/attn_blocks.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
import math

import torch


+ 2
- 1
modelscope/models/cv/video_single_object_tracking/models/layers/head.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
import torch
import torch.nn as nn



+ 2
- 1
modelscope/models/cv/video_single_object_tracking/models/layers/patch_embed.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
import torch.nn as nn
from timm.models.layers import to_2tuple



+ 2
- 1
modelscope/models/cv/video_single_object_tracking/models/ostrack/base_backbone.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
import torch.nn as nn
from timm.models.layers import to_2tuple



+ 2
- 1
modelscope/models/cv/video_single_object_tracking/models/ostrack/ostrack.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
import torch
from torch import nn



+ 2
- 1
modelscope/models/cv/video_single_object_tracking/models/ostrack/utils.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
import torch




+ 2
- 1
modelscope/models/cv/video_single_object_tracking/models/ostrack/vit_ce.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
from functools import partial

import torch


+ 2
- 1
modelscope/models/cv/video_single_object_tracking/tracker/ostrack.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
import torch

from modelscope.models.cv.video_single_object_tracking.config.ostrack import \


+ 2
- 1
modelscope/models/cv/video_single_object_tracking/utils/utils.py View File

@@ -1,4 +1,5 @@
# The implementation is based on OSTrack, available at https://github.com/botaoye/OSTrack/
# The implementation is adopted from OSTrack,
# made publicly available under the MIT License at https://github.com/botaoye/OSTrack/
import math
from typing import Optional



+ 1
- 0
modelscope/pipelines/cv/video_single_object_tracking_pipeline.py View File

@@ -1,3 +1,4 @@
# Copyright (c) Alibaba, Inc. and its affiliates.
import os.path as osp
from typing import Any, Dict



Loading…
Cancel
Save