Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
I
Im
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
文档服务地址:
http://47.92.0.57:3000/
周报索引地址:
http://47.92.0.57:3000/s/NruNXRYmV
Open sidebar
王肇一
Im
Commits
f07209f5
Commit
f07209f5
authored
Jan 09, 2020
by
王肇一
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
channel 1
parent
ac7595c5
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
predict.py
predict.py
+1
-1
train.py
train.py
+1
-1
No files found.
predict.py
View file @
f07209f5
...
...
@@ -102,7 +102,7 @@ if __name__ == "__main__":
in_files
=
args
.
input
out_files
=
get_output_filenames
(
args
)
net
=
UNet
(
n_channels
=
3
,
n_classes
=
1
)
net
=
UNet
(
n_channels
=
1
,
n_classes
=
1
)
logging
.
info
(
"Loading model {}"
.
format
(
args
.
model
))
...
...
train.py
View file @
f07209f5
...
...
@@ -146,7 +146,7 @@ if __name__ == '__main__':
# - For 1 class and background, use n_classes=1
# - For 2 classes, use n_classes=1
# - For N > 2 classes, use n_classes=N
net
=
UNet
(
n_channels
=
3
,
n_classes
=
1
)
net
=
UNet
(
n_channels
=
1
,
n_classes
=
1
)
logging
.
info
(
f
'Network:
\n
'
f
'
\t
{net.n_channels} input channels
\n
'
f
'
\t
{net.n_classes} output channels (classes)
\n
'
...
...
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