add referring tracker tool for track anything #67
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
作者们你们好！我是书生·浦语大模型训练营第二期的学员~
在项目阶段，我想对agent进行一点研究，结合我自己的研究方向（目标跟踪），我发现agentlego有检测，分割的一些tool，但是没有跟踪相关的。如果用户输入一个视频，并且希望LLM对其中的某个物品进行跟踪，我写的这个tracking tool或许可以派上用场~
技术手段是通过GLIP或SAM进行检测或分割，然后用类似ByteTtrack的算法进行跟踪。为了支持视频（或图片序列文件夹）的输入，我也更改了
types.py中的一些内容。更改的部分是
agentlego/agentlego/tools/tracking和types.py.以下是示例以及运行的结果，LLM采用InternLM-1.8b或7b.
Please segment and track the laptop and cup in the videotest.mp4``