Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
YOLO-World_pytorch
Commits
e7784147
Commit
e7784147
authored
May 31, 2024
by
luopl
Browse files
Update README.md
parent
e9cee049
Pipeline
#1057
canceled with stages
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
0 deletions
+3
-0
README.md
README.md
+3
-0
No files found.
README.md
View file @
e7784147
...
...
@@ -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
)
注意:模型配置文件与权重文件应一一对应
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment