Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
2301_77200941
mindspore
提交
a99bc1a9
M
mindspore
项目概览
2301_77200941
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
提交
a99bc1a9
编写于
8月 29, 2020
作者:
G
gengdongjie
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
bugfix on argpasr for bool
上级
1821e98e
变更
16
隐藏空白更改
内联
并排
Showing
16 changed file
with
12 addition
and
11 deletion
+12
-11
model_zoo/official/cv/maskrcnn/README.md
model_zoo/official/cv/maskrcnn/README.md
+2
-2
model_zoo/official/cv/maskrcnn/eval.py
model_zoo/official/cv/maskrcnn/eval.py
+1
-1
model_zoo/official/cv/maskrcnn/src/config.py
model_zoo/official/cv/maskrcnn/src/config.py
+2
-2
model_zoo/official/cv/maskrcnn/src/maskrcnn/__init__.py
model_zoo/official/cv/maskrcnn/src/maskrcnn/__init__.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/anchor_generator.py
...zoo/official/cv/maskrcnn/src/maskrcnn/anchor_generator.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/bbox_assign_sample.py
...o/official/cv/maskrcnn/src/maskrcnn/bbox_assign_sample.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/bbox_assign_sample_stage2.py
...ial/cv/maskrcnn/src/maskrcnn/bbox_assign_sample_stage2.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/fpn_neck.py
model_zoo/official/cv/maskrcnn/src/maskrcnn/fpn_neck.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/mask_rcnn_r50.py
model_zoo/official/cv/maskrcnn/src/maskrcnn/mask_rcnn_r50.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/proposal_generator.py
...o/official/cv/maskrcnn/src/maskrcnn/proposal_generator.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/rcnn_cls.py
model_zoo/official/cv/maskrcnn/src/maskrcnn/rcnn_cls.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/rcnn_mask.py
model_zoo/official/cv/maskrcnn/src/maskrcnn/rcnn_mask.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/resnet50.py
model_zoo/official/cv/maskrcnn/src/maskrcnn/resnet50.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/roi_align.py
model_zoo/official/cv/maskrcnn/src/maskrcnn/roi_align.py
+0
-0
model_zoo/official/cv/maskrcnn/src/maskrcnn/rpn.py
model_zoo/official/cv/maskrcnn/src/maskrcnn/rpn.py
+0
-0
model_zoo/official/cv/maskrcnn/train.py
model_zoo/official/cv/maskrcnn/train.py
+7
-6
未找到文件。
model_zoo/official/cv/maskrcnn/README.md
浏览文件 @
a99bc1a9
...
@@ -50,7 +50,7 @@ MaskRcnn is a two-stage target detection network,This network uses a region prop
...
@@ -50,7 +50,7 @@ MaskRcnn is a two-stage target detection network,This network uses a region prop
```
shell
```
shell
.
.
└─
MaskRcnn
└─
maskrcnn
├─README.md
├─README.md
├─scripts
├─scripts
├─run_download_process_data.sh
├─run_download_process_data.sh
...
@@ -58,7 +58,7 @@ MaskRcnn is a two-stage target detection network,This network uses a region prop
...
@@ -58,7 +58,7 @@ MaskRcnn is a two-stage target detection network,This network uses a region prop
├─run_train.sh
├─run_train.sh
└─run_eval.sh
└─run_eval.sh
├─src
├─src
├─
MaskR
cnn
├─
maskr
cnn
├─__init__.py
├─__init__.py
├─anchor_generator.py
├─anchor_generator.py
├─bbox_assign_sample.py
├─bbox_assign_sample.py
...
...
model_zoo/official/cv/maskrcnn/eval.py
浏览文件 @
a99bc1a9
...
@@ -24,7 +24,7 @@ from mindspore import context, Tensor
...
@@ -24,7 +24,7 @@ from mindspore import context, Tensor
from
mindspore.train.serialization
import
load_checkpoint
,
load_param_into_net
from
mindspore.train.serialization
import
load_checkpoint
,
load_param_into_net
import
mindspore.dataset.engine
as
de
import
mindspore.dataset.engine
as
de
from
src.
MaskR
cnn.mask_rcnn_r50
import
Mask_Rcnn_Resnet50
from
src.
maskr
cnn.mask_rcnn_r50
import
Mask_Rcnn_Resnet50
from
src.config
import
config
from
src.config
import
config
from
src.dataset
import
data_to_mindrecord_byte_image
,
create_maskrcnn_dataset
from
src.dataset
import
data_to_mindrecord_byte_image
,
create_maskrcnn_dataset
from
src.util
import
coco_eval
,
bbox2result_1image
,
results2json
,
get_seg_masks
from
src.util
import
coco_eval
,
bbox2result_1image
,
results2json
,
get_seg_masks
...
...
model_zoo/official/cv/maskrcnn/src/config.py
浏览文件 @
a99bc1a9
...
@@ -141,8 +141,8 @@ config = ed({
...
@@ -141,8 +141,8 @@ config = ed({
"keep_checkpoint_max"
:
12
,
"keep_checkpoint_max"
:
12
,
"save_checkpoint_path"
:
"./checkpoint"
,
"save_checkpoint_path"
:
"./checkpoint"
,
"mindrecord_dir"
:
"/home/m
xw/mask_rcnn/scripts
/MindRecord_COCO2017_Train"
,
"mindrecord_dir"
:
"/home/m
ask_rcnn
/MindRecord_COCO2017_Train"
,
"coco_root"
:
"/home/m
xw
/coco2017/"
,
"coco_root"
:
"/home/m
ask_rcnn
/coco2017/"
,
"train_data_type"
:
"train2017"
,
"train_data_type"
:
"train2017"
,
"val_data_type"
:
"val2017"
,
"val_data_type"
:
"val2017"
,
"instance_set"
:
"annotations/instances_{}.json"
,
"instance_set"
:
"annotations/instances_{}.json"
,
...
...
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/__init__.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/__init__.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/anchor_generator.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/anchor_generator.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/bbox_assign_sample.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/bbox_assign_sample.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/bbox_assign_sample_stage2.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/bbox_assign_sample_stage2.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/fpn_neck.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/fpn_neck.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/mask_rcnn_r50.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/mask_rcnn_r50.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/proposal_generator.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/proposal_generator.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/rcnn_cls.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/rcnn_cls.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/rcnn_mask.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/rcnn_mask.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/resnet50.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/resnet50.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/roi_align.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/roi_align.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/src/
MaskR
cnn/rpn.py
→
model_zoo/official/cv/maskrcnn/src/
maskr
cnn/rpn.py
浏览文件 @
a99bc1a9
文件已移动
model_zoo/official/cv/maskrcnn/train.py
浏览文件 @
a99bc1a9
...
@@ -18,6 +18,7 @@
...
@@ -18,6 +18,7 @@
import
os
import
os
import
argparse
import
argparse
import
random
import
random
import
ast
import
numpy
as
np
import
numpy
as
np
import
mindspore.common.dtype
as
mstype
import
mindspore.common.dtype
as
mstype
...
@@ -30,7 +31,7 @@ from mindspore.train.serialization import load_checkpoint, load_param_into_net
...
@@ -30,7 +31,7 @@ from mindspore.train.serialization import load_checkpoint, load_param_into_net
from
mindspore.nn
import
SGD
from
mindspore.nn
import
SGD
import
mindspore.dataset.engine
as
de
import
mindspore.dataset.engine
as
de
from
src.
MaskR
cnn.mask_rcnn_r50
import
Mask_Rcnn_Resnet50
from
src.
maskr
cnn.mask_rcnn_r50
import
Mask_Rcnn_Resnet50
from
src.network_define
import
LossCallBack
,
WithLossCell
,
TrainOneStepCell
,
LossNet
from
src.network_define
import
LossCallBack
,
WithLossCell
,
TrainOneStepCell
,
LossNet
from
src.config
import
config
from
src.config
import
config
from
src.dataset
import
data_to_mindrecord_byte_image
,
create_maskrcnn_dataset
from
src.dataset
import
data_to_mindrecord_byte_image
,
create_maskrcnn_dataset
...
@@ -41,11 +42,11 @@ np.random.seed(1)
...
@@ -41,11 +42,11 @@ np.random.seed(1)
de
.
config
.
set_seed
(
1
)
de
.
config
.
set_seed
(
1
)
parser
=
argparse
.
ArgumentParser
(
description
=
"MaskRcnn training"
)
parser
=
argparse
.
ArgumentParser
(
description
=
"MaskRcnn training"
)
parser
.
add_argument
(
"--only_create_dataset"
,
type
=
boo
l
,
default
=
False
,
help
=
"If set it true, only create "
parser
.
add_argument
(
"--only_create_dataset"
,
type
=
ast
.
literal_eva
l
,
default
=
False
,
help
=
"If set it true, only create "
"Mindrecord, default is false."
)
"Mindrecord, default is false."
)
parser
.
add_argument
(
"--run_distribute"
,
type
=
boo
l
,
default
=
False
,
help
=
"Run distribute, default is false."
)
parser
.
add_argument
(
"--run_distribute"
,
type
=
ast
.
literal_eva
l
,
default
=
False
,
help
=
"Run distribute, default is false."
)
parser
.
add_argument
(
"--do_train"
,
type
=
boo
l
,
default
=
True
,
help
=
"Do train or not, default is true."
)
parser
.
add_argument
(
"--do_train"
,
type
=
ast
.
literal_eva
l
,
default
=
True
,
help
=
"Do train or not, default is true."
)
parser
.
add_argument
(
"--do_eval"
,
type
=
boo
l
,
default
=
False
,
help
=
"Do eval or not, default is false."
)
parser
.
add_argument
(
"--do_eval"
,
type
=
ast
.
literal_eva
l
,
default
=
False
,
help
=
"Do eval or not, default is false."
)
parser
.
add_argument
(
"--dataset"
,
type
=
str
,
default
=
"coco"
,
help
=
"Dataset, default is coco."
)
parser
.
add_argument
(
"--dataset"
,
type
=
str
,
default
=
"coco"
,
help
=
"Dataset, default is coco."
)
parser
.
add_argument
(
"--pre_trained"
,
type
=
str
,
default
=
""
,
help
=
"Pretrain file path."
)
parser
.
add_argument
(
"--pre_trained"
,
type
=
str
,
default
=
""
,
help
=
"Pretrain file path."
)
parser
.
add_argument
(
"--device_id"
,
type
=
int
,
default
=
0
,
help
=
"Device id, default is 0."
)
parser
.
add_argument
(
"--device_id"
,
type
=
int
,
default
=
0
,
help
=
"Device id, default is 0."
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录