Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
王党生
classification
提交
c7a55d30
classification
项目概览
王党生
/
classification
与 Fork 源项目一致
Fork自
DataBall / classification
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
classification
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
c7a55d30
编写于
2月 13, 2021
作者:
DataBall
🚴🏻
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update
上级
6cf3aaf5
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
9 addition
and
8 deletion
+9
-8
README.md
README.md
+4
-3
inference.py
inference.py
+2
-2
read_datasets.py
read_datasets.py
+3
-3
未找到文件。
README.md
浏览文件 @
c7a55d30
...
@@ -10,11 +10,12 @@
...
@@ -10,11 +10,12 @@
*
PyTorch >= 1.5.1
*
PyTorch >= 1.5.1
## 数据集
## 数据集
采用"Stanford Dogs Dataset"数据集官方地址:http://vision.stanford.edu/aditya86/ImageNetDogs/,且分为训练和测试两部分。
*
采用"Stanford Dogs Dataset"数据集官方地址:http://vision.stanford.edu/aditya86/ImageNetDogs/,且分为训练和测试两部分。
本文将该数据集的标注文件改为xml格式,可以通过运行read_datasests.py,可以对数据的标注信息进行解析可视化。
*
本文将该数据集的标注文件更改为xml格式,
[
数据集下载地址(百度网盘 Password: ks87 )
](
https://pan.baidu.com/s/1tT0wF4N2I9p5JDfCwtM1CQ
)
*
通过运行read_datasests.py,可以对数据的标注信息进行解析可视化。
## 预训练模型
## 预训练模型
*
[预训练模型下载地址
: ~
*
[
预训练模型下载地址
(百度网盘 Password: ks87 )
](
https://pan.baidu.com/s/1tT0wF4N2I9p5JDfCwtM1CQ
)
## 项目使用方法
## 项目使用方法
...
...
inference.py
浏览文件 @
c7a55d30
...
@@ -41,7 +41,7 @@ def get_xml_msg(path):
...
@@ -41,7 +41,7 @@ def get_xml_msg(path):
if
__name__
==
"__main__"
:
if
__name__
==
"__main__"
:
parser
=
argparse
.
ArgumentParser
(
description
=
' Project Classification Test'
)
parser
=
argparse
.
ArgumentParser
(
description
=
' Project Classification Test'
)
parser
.
add_argument
(
'--test_model'
,
type
=
str
,
default
=
'./model_exp/2021-02-09_06-32-32/
model
_epoch-627.pth'
,
parser
.
add_argument
(
'--test_model'
,
type
=
str
,
default
=
'./model_exp/2021-02-09_06-32-32/
resnet50
_epoch-627.pth'
,
help
=
'test_model'
)
# 模型路径
help
=
'test_model'
)
# 模型路径
parser
.
add_argument
(
'--model'
,
type
=
str
,
default
=
'resnet_50'
,
parser
.
add_argument
(
'--model'
,
type
=
str
,
default
=
'resnet_50'
,
help
=
'model : resnet_18,resnet_34,resnet_50,resnet_101,resnet_152'
)
# 模型类型
help
=
'model : resnet_18,resnet_34,resnet_50,resnet_101,resnet_152'
)
# 模型类型
...
@@ -177,7 +177,7 @@ if __name__ == "__main__":
...
@@ -177,7 +177,7 @@ if __name__ == "__main__":
dict_r
[
doc
]
+=
1
dict_r
[
doc
]
+=
1
cv2
.
destroyAllWindows
()
cv2
.
destroyAllWindows
()
# Top1 的
每类预测精确度。
# Top1 的每类预测精确度。
print
(
'
\n
-----------------------------------------------
\n
'
)
print
(
'
\n
-----------------------------------------------
\n
'
)
acc_list
=
[]
acc_list
=
[]
for
idx
,
doc
in
enumerate
(
sorted
(
os
.
listdir
(
ops
.
test_path
),
key
=
lambda
x
:
int
(
x
.
split
(
'-'
)[
0
]),
reverse
=
False
)):
for
idx
,
doc
in
enumerate
(
sorted
(
os
.
listdir
(
ops
.
test_path
),
key
=
lambda
x
:
int
(
x
.
split
(
'-'
)[
0
]),
reverse
=
False
)):
...
...
read_datasets.py
浏览文件 @
c7a55d30
#-*-coding:utf-8-*-
#-*-coding:utf-8-*-
# date:2020-02-08
# date:2020-02-08
#
A
uthor: Eric.Lee
#
a
uthor: Eric.Lee
#
#
function: read datasets label files
# function: read datasets label files
import
os
import
os
import
cv2
import
cv2
...
@@ -56,7 +56,7 @@ if __name__ == "__main__":
...
@@ -56,7 +56,7 @@ if __name__ == "__main__":
cv2
.
putText
(
img_
,
(
'index : '
+
str
(
idx
)),
(
5
,
img_
.
shape
[
0
]
-
5
),
cv2
.
FONT_HERSHEY_PLAIN
,
1.8
,
(
255
,
255
,
0
),
6
)
cv2
.
putText
(
img_
,
(
'index : '
+
str
(
idx
)),
(
5
,
img_
.
shape
[
0
]
-
5
),
cv2
.
FONT_HERSHEY_PLAIN
,
1.8
,
(
255
,
255
,
0
),
6
)
cv2
.
putText
(
img_
,
(
'index : '
+
str
(
idx
)),
(
5
,
img_
.
shape
[
0
]
-
5
),
cv2
.
FONT_HERSHEY_PLAIN
,
1.8
,
(
255
,
60
,
255
),
2
)
cv2
.
putText
(
img_
,
(
'index : '
+
str
(
idx
)),
(
5
,
img_
.
shape
[
0
]
-
5
),
cv2
.
FONT_HERSHEY_PLAIN
,
1.8
,
(
255
,
60
,
255
),
2
)
xml_
=
path
+
doc_
+
"/"
+
f_
.
strip
(
'.jpg'
).
strip
(
'.png'
)
+
'.xml'
xml_
=
path
+
doc_
+
"/"
+
f_
.
replace
(
".jpg"
,
".xml"
).
replace
(
".png"
,
".xml"
)
list_x
=
get_xml_msg
(
xml_
)
# 获取 xml 文件 的 object
list_x
=
get_xml_msg
(
xml_
)
# 获取 xml 文件 的 object
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录