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
```
shell
.
└─
MaskRcnn
└─
maskrcnn
├─README.md
├─scripts
├─run_download_process_data.sh
...
...
@@ -58,7 +58,7 @@ MaskRcnn is a two-stage target detection network,This network uses a region prop
├─run_train.sh
└─run_eval.sh
├─src
├─
MaskR
cnn
├─
maskr
cnn
├─__init__.py
├─anchor_generator.py
├─bbox_assign_sample.py
...
...
model_zoo/official/cv/maskrcnn/eval.py
浏览文件 @
a99bc1a9
...
...
@@ -24,7 +24,7 @@ from mindspore import context, Tensor
from
mindspore.train.serialization
import
load_checkpoint
,
load_param_into_net
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.dataset
import
data_to_mindrecord_byte_image
,
create_maskrcnn_dataset
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({
"keep_checkpoint_max"
:
12
,
"save_checkpoint_path"
:
"./checkpoint"
,
"mindrecord_dir"
:
"/home/m
xw/mask_rcnn/scripts
/MindRecord_COCO2017_Train"
,
"coco_root"
:
"/home/m
xw
/coco2017/"
,
"mindrecord_dir"
:
"/home/m
ask_rcnn
/MindRecord_COCO2017_Train"
,
"coco_root"
:
"/home/m
ask_rcnn
/coco2017/"
,
"train_data_type"
:
"train2017"
,
"val_data_type"
:
"val2017"
,
"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 @@
import
os
import
argparse
import
random
import
ast
import
numpy
as
np
import
mindspore.common.dtype
as
mstype
...
...
@@ -30,7 +31,7 @@ from mindspore.train.serialization import load_checkpoint, load_param_into_net
from
mindspore.nn
import
SGD
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.config
import
config
from
src.dataset
import
data_to_mindrecord_byte_image
,
create_maskrcnn_dataset
...
...
@@ -41,11 +42,11 @@ np.random.seed(1)
de
.
config
.
set_seed
(
1
)
parser
=
argparse
.
ArgumentParser
(
description
=
"MaskRcnn training"
)
parser
.
add_argument
(
"--only_create_dataset"
,
type
=
boo
l
,
default
=
False
,
help
=
"If set it true, only create "
"Mindrecord, default is false."
)
parser
.
add_argument
(
"--run_distribute"
,
type
=
boo
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_eval"
,
type
=
boo
l
,
default
=
False
,
help
=
"Do eval or not, default is false."
)
parser
.
add_argument
(
"--only_create_dataset"
,
type
=
ast
.
literal_eva
l
,
default
=
False
,
help
=
"If set it true, only create "
"Mindrecord, 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
=
ast
.
literal_eva
l
,
default
=
True
,
help
=
"Do train or not, default is true."
)
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
(
"--pre_trained"
,
type
=
str
,
default
=
""
,
help
=
"Pretrain file path."
)
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.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录