Commit e7784147 authored by luopl's avatar luopl
Browse files

Update README.md

parent e9cee049
Pipeline #1057 canceled with stages
......@@ -89,6 +89,7 @@ pip install -v -e .
| [GQA](https://downloads.cs.stanford.edu/nlp/data/gqa/images.zip) | [final_mixed_train_no_coco.json](https://huggingface.co/GLIPModel/GLIP/tree/main/mdetr_annotations) |
| [Flickr30k](https://opendatalab.com/OpenDataLab/Flickr30k) | [final_flickr_separateGT_train.json](https://huggingface.co/GLIPModel/GLIP/tree/main/mdetr_annotations) |
| [COCO val2017](https://opendatalab.com/OpenDataLab/COCO_2017) | [lvis_v1_minival_inserted_image_name.json](https://huggingface.co/GLIPModel/GLIP/blob/main/lvis_v1_minival_inserted_image_name.json) |
将所有数据下载并放置于data文件夹下,数据目录如下:
注意:lvis_v1_minival_inserted_image_name.json放于coco/lvis/文件夹下,其余数据标注文件放于各数据文件的annotations文件夹下
......@@ -112,6 +113,7 @@ pip install -v -e .
```
## 训练
首先下载clip-vit模型文件,放于openai目录下:
[clip-vit-large-patch14-336下载](https://huggingface.co/openai/clip-vit-large-patch14-336)
[clip-vit-base-patch32下载](https://huggingface.co/openai/clip-vit-base-patch32)
### 单机单卡
......@@ -134,6 +136,7 @@ chmod +x tools/dist_train.sh
```
## 推理
根据以下链接下载想要的模型权重文件,放到weights文件夹下:
[模型权重文件下载](https://hf-mirror.com/wondervictor/YOLO-World/tree/main)
注意:模型配置文件与权重文件应一一对应
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment