API文档
项目结构
- run: 启动整个程序运行的逻辑，包括通过服务器提供服务、训练、测试、评估等
- reid_pipeline: Re-id流程主体逻辑
- models: 模型具体实现，包括各模型和用于管理模型的类
- database_manager: 数据库管理，包括数据增删查改等
- utils: 工具类，包括各种工具函数、工具类
基本API
run
reid_pipeline
pipeline.py
class Pipeline:
def __init__(self):
pass
def __call__(self, *args, **kwargs):
pass
def spot_object_from_image(self, image):
pass
def spot_object_from_video(self, video):
pass
def get_embedding(self, objects):
pass
def submit_result(self, embeddings):
pass
reid_data_manager.py
class DetectedObject: #封装检测到的目标的数据
def __init__(self, cam_id, img, bike_person_img, score, cls_id, center):
self.cam_id = cam_id
self.img = img
self.bike_person_img = bike_person_img
self.score = score
self.cls_id = cls_id
self.center = center
self.embedding: torch.Tensor = None
self.time = time.time() #时间戳除以1000，单位为秒
数据库结构
sever_pipeline.py
class ServerPipeline:
def insert_new_data_from_img(self, img, cam_id):
# 主功能一：将监控图像中的自行车插入数据库
def query_img(self, img, top_k=10):
# 主功能二：接受用户查询，返回前top_k辆相似的自行车，并返回自行车出现的记录
MySQL
| 0 |
1 |
2 |
3 |
4 |
5 |
6 |
| id |
bicycle_id |
camera_id |
start_time |
end_time |
location_desc |
img_path |
| Field |
Type |
Null |
Key |
Default |
Extra |
| id |
bigint(20) unsigned |
NO |
PRI |
NULL |
auto_increment |
| bicycle_id |
bigint(20) unsigned |
YES |
MUL |
NULL |
|
| camera_id |
int(10) unsigned |
YES |
|
NULL |
|
| start_time |
bigint(20) |
YES |
|
NULL |
|
| end_time |
bigint(20) |
YES |
|
NULL |
|
| location_desc |
varchar(50) |
YES |
|
NULL |
|
| img_path |
varchar(100) |
YES |
|
NULL |
|
milvus