2024“大运河杯”数据开发应用创新大赛——城市治理 [Datawhale AI 夏令营]

---

赛题

赛题源自 2024“大运河杯”数据开发应用创新大赛——城市治理。
简单而言，利用已有标注的违规（违规行为包括垃圾桶满溢、机动车违停、非机动车违停等）视频数据来有效对未标注视频标注，赛题提供了以下三个数据集：
训练集（有标注）、
训练集（无标注）、
测试集。

数据处理

由于视频数据为mp4格式，标注文件为json格式，每个视频对应一个json文件，需要将mp4等视频有效文件转化为图片形式保存（下面代码中给出作为train数据具体的注释和注意事项，val数据也类似）:

import os, sys
import cv2, glob, json
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

anno_df = pd.read_json(anno_path)
# 读取json文件
cap = cv2.VideoCapture(video_path)
# 同时读取视频文件

### 这里需要自定义读取多少内容作为trai样本
frame_idx = 0 
while True:
	ret, frame = cap.read()
	# 帧图片流
	if not ret:
	   break
	
	img_height, img_width = frame.shape[:2