diff --git a/model_examples/DiffusionDrive/README.md b/model_examples/DiffusionDrive/README.md index 2e492869bc1b6eb31b746c2c6f3edbc4b49b7736..54c860ebbfbbba77d18d98f1c3cd97aa60e9e0fc 100644 --- a/model_examples/DiffusionDrive/README.md +++ b/model_examples/DiffusionDrive/README.md @@ -167,7 +167,7 @@ ln -s $DATA_PATH/v1.0-trainval ./data/nuscenes/v1.0-trainval 运行数据预处理脚本生成DiffusionDrive模型训练需要的pkl文件与初始锚框(预处理耗时约5h左右) ```bash -python ./migrate_to_ascened/preprocess.py +python ./migrate_to_ascend/preprocess.py ``` ## 下载权重 diff --git a/model_examples/DiffusionDrive/migrate_to_ascend/requirements.txt b/model_examples/DiffusionDrive/migrate_to_ascend/requirements.txt index 1070dbe86c4ee54a58ad855f57ad5e9d86e5dc05..a8996082ece211a06f21fdbe63bb4fd805e1826b 100644 --- a/model_examples/DiffusionDrive/migrate_to_ascend/requirements.txt +++ b/model_examples/DiffusionDrive/migrate_to_ascend/requirements.txt @@ -14,6 +14,6 @@ scikit-learn==1.3.0 IPython==8.12 einops==0.8.1 ml-dtypes -opencv-python-headless==4.6.0.66 +opencv-python-headless==4.8.1.78 diffusers torchvision==0.16.0 \ No newline at end of file