CVAT项目中的SDK导出数据集问题分析与解决方案

2025-05-16 00:53:14作者：宣海椒Queenly

问题背景

在CVAT（计算机视觉标注工具）项目中，用户在使用cvat_sdk 2.28.0版本时遇到了一个关键问题：通过export_dataset方法导出的数据集文件实际上是一个包含HTML内容的文本文件，而非预期的ZIP压缩包。这个问题在Windows 11和Ubuntu 22.04系统上均能复现。

问题现象分析

当用户执行以下代码时：

task = client.tasks.retrieve(TASK_ID)
task.export_dataset(
    format_name="COCO 1.0",
    filename="example.zip",
    include_images=True,
    location=Location.LOCAL,
)

得到的"example.zip"文件实际上是一个HTML文件，内容如下：

<!DOCTYPE html>
<html>
    <body>
        <input type="hidden" id="http" name="http" value="5000">
        <input type="hidden" id="https" name="https" value="5001">
        <input type="hidden" id="prefer_https" name="prefer_https" value="true">
    </body>
    <script type="text/javascript">
        var protocol="https:";
        var port=5001;
        var URL=protocol+"//"+location.hostname+":"+port+location.pathname+location.search;
        location.replace(URL);
    </script>
</html>

根本原因

经过深入分析，发现问题的根源在于HTTPS重定向机制：

在CVAT 2.28.0版本中，SDK生成的下载链接默认为HTTP协议
当服务器配置强制使用HTTPS时，会返回这个重定向HTML页面
SDK未能正确处理这个重定向响应，导致将HTML内容保存为文件

解决方案

针对这个问题，有以下几种解决方法：

方法一：修改服务器配置

在CVAT服务器配置中明确设置基础URL为HTTPS协议，确保SDK生成的链接直接使用HTTPS：

# 在settings.py或production.py中添加
CVAT_BASE_URL = "https://your-cvat-domain.com"

方法二：修改SDK代码

对于无法修改服务器配置的情况，可以在SDK使用端进行修改：

from cvat_sdk.core.downloading import Downloader

# 重写prepare_url方法
def prepare_url(url: str) -> str:
    return url.replace("http://", "https://")

Downloader.prepare_url = prepare_url