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
yolov5_pytorch
Commits
daf7358d
Commit
daf7358d
authored
Jul 11, 2024
by
dengjb
Browse files
update
parent
207d5fb2
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
18 additions
and
9 deletions
+18
-9
README.md
README.md
+18
-9
icon.png
icon.png
+0
-0
No files found.
README.md
View file @
daf7358d
...
...
@@ -62,15 +62,16 @@ pip3 install -r requirements.txt
## 数据集
COCO2017(在网络良好的情况下,如果没有下载数据集,程序会默认在线下载数据集)
[
训练数据
](
http://images.cocodataset.org/zips/train2017.zip
)
[
验证数据
](
http://images.cocodataset.org/zips/val2017.zip
)
[
测试数据
](
http://images.cocodataset.org/zips/test2017.zip
)
[
标签数据
](
https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017labels.zip
)
在本测试中可以使用COCO2017数据集。
-
数据集快速下载中心:
-
[
SCNet AIDatasets
](
http://113.200.138.88:18080/aidatasets
)
-
数据集快速通道下载地址:
-
[
数据集快速下载地址
](
http://113.200.138.88:18080/aidatasets/coco2017
)
-
官方下载地址
-
[
训练数据
](
http://images.cocodataset.org/zips/train2017.zip
)
-
[
验证数据
](
http://images.cocodataset.org/zips/val2017.zip
)
-
[
测试数据
](
http://images.cocodataset.org/zips/test2017.zip
)
-
[
标签数据
](
https://github.com/ultralytics/yolov5/releases/download/v1.0/coco2017labels.zip
)
数据集的目录结构如下:
...
...
@@ -141,6 +142,14 @@ python3 -m torch.distributed.launch --nproc_per_node 4 --nnodes 2 --node_rank 1
```
## 推理
-
可使用官方模型权重进行推理,也可使用自己训练模型权重进行推理
-
模型权重快速下载中心
-
[
SCNet AIModels
](
http://113.200.138.88:18080/aimodels
)
-
模型权重快速通道下载地址
-
[
模型权重快速下载地址
](
http://113.200.138.88:18080/aimodels/findsource-dependency/yolov5_pytorch
)
-
官方下载地址:
-
[
下载地址:yolov5s.pt
](
https://github.com/ultralytics/yolov5/releases/download/v7.0/yolov5s.pt
)
-
这里以官方模型推理举例
#### 单卡推理
...
...
icon.png
0 → 100644
View file @
daf7358d
77.3 KB
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