Datawhale 第五期CV task1运行baseline-CSDN博客

本文链接：https://blog.csdn.net/qq_51406661/article/details/141556364

Step1.报名参赛：

1.登录/注册赛事平台

2. 修改昵称并进行实名认证

3.报名参赛

4.修改队伍名称

赛事平台连接：https://www.marsbigdata.com/competition/details?id=3839107548872

进入平台后，依照指引进行注册和实名认证即可参与比赛。值得注意的是了解渠道要填写Datawhale/AI 夏令营/夏令营

Step2.领取厚德云支持的GPU在线算力

本次学习免费算力由厚德云提供支持，为每位学习者提供50元算力（仅限参与夏令营的学习者）

1.注册厚德云平台（点击跳转注册）： https://portal.houdeyun.cn/register?from=Datawhale

2.具体步骤图：

Step3.30分钟快速跑通baseline

1.删除除了data和public以外的所有文件夹，然后进入终端

2.输入命令

apt install git-lfs
git lfs install
git clone https://www.modelscope.cn/datasets/Datawhale/AI_Camp5_baseline_CV.git

3.等待出现AI_Camp5_baseline_CV,一键运行baseline（大约需要25分钟），耐心等待

这里的Baseline会出现一个问题，此处代码段，baseline会读取一个最佳权重，但是上文并没有保存，所以把那句代码删除即可

from ultralytics import YOLO
import glob

for path in glob.glob('测试集/*.mp4'):
    submit_json = []
    results = model(path, conf=0.05, imgsz=1080,  verbose=False)
    for idx, result in enumerate(results):
        boxes = result.boxes  # Boxes object for bounding box outputs
        masks = result.masks  # Masks object for segmentation masks outputs
        keypoints = result.keypoints  # Keypoints object for pose outputs
        probs = result.probs  # Probs object for classification outputs
        obb = result.obb  # Oriented boxes object for OBB outputs

        if len(boxes.cls) == 0:
            continue
        
        xywh = boxes.xyxy.data.cpu().numpy().round()
        cls = boxes.cls.data.cpu().numpy().round()
        conf = boxes.conf.data.cpu().numpy()
        for i, (ci, xy, confi) in enumerate(zip(cls, xywh, conf)):
            submit_json.append(
                {
                    'frame_id': idx,
                    'event_id': i+1,
                    'category': category_labels[int(ci)],
                    'bbox': list([int(x) for x in xy]),
                    "confidence": float(confi)
                }
            )

    with open('./result/' + path.split('/')[-1][:-4] + '.json', 'w', encoding='utf-8') as up:
        json.dump(submit_json, up, indent=4, ensure_ascii=False)

4.下载出现的result.zip

5.记得GPU使用完成后，关机~