# 一种基于OSTrack的ava数据集辅助标注脚本

**Repository Path**: gysen/script_for_label

## Basic Information

- **Project Name**: 一种基于OSTrack的ava数据集辅助标注脚本
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-05-14
- **Last Updated**: 2024-05-14

## Categories & Tags

**Categories**: Uncategorized

**Tags**: Python, torch, modelscope

## README

可以实现简易的视频辅助标注，需要先在源代码中给出video_path（视频路径）以及init_bbox的坐标值，脚本会自动将跟踪结果写入output.csv文件
在venv中（一定要在虚拟环境中）运行以下代码安装依赖
建议按照官网教程安装cuda之后再运行脚本，不然系统自动使用cpu风扇狂转
```shell
pip install -r requirements.txt
```
当遇到清华源较慢的情况可以还原至北京外国语大学

在使用该脚本标注数据集时
<br>
注意更改以下信息<br>

```python
video_path = 'tracking_result.avi'  # 替换为你的视频文件路径
init_bbox = [414, 343, 514, 449]  # 初始框 [x1, y1, x2, y2] 以左上角为[0,0]
pig_id = 0 #当前视频中标注的第几个猪
action_id = 0 
'''
暂定walking:0
standing:1
lying:2
'''
```