Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
O
opencv
Project
Project
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
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
opencv
Commits
f29bc220
Commit
f29bc220
authored
Jan 15, 2020
by
ashishiva3@gmail.com
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Convolutional padding parsing modified
parent
3f27f8cf
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
10 additions
and
4 deletions
+10
-4
darknet_io.cpp
modules/dnn/src/darknet/darknet_io.cpp
+4
-3
test_darknet_importer.cpp
modules/dnn/test/test_darknet_importer.cpp
+6
-1
No files found.
modules/dnn/src/darknet/darknet_io.cpp
View file @
f29bc220
...
@@ -556,6 +556,7 @@ namespace cv {
...
@@ -556,6 +556,7 @@ namespace cv {
{
{
int
kernel_size
=
getParam
<
int
>
(
layer_params
,
"size"
,
-
1
);
int
kernel_size
=
getParam
<
int
>
(
layer_params
,
"size"
,
-
1
);
int
pad
=
getParam
<
int
>
(
layer_params
,
"pad"
,
0
);
int
pad
=
getParam
<
int
>
(
layer_params
,
"pad"
,
0
);
int
padding
=
getParam
<
int
>
(
layer_params
,
"padding"
,
0
);
int
stride
=
getParam
<
int
>
(
layer_params
,
"stride"
,
1
);
int
stride
=
getParam
<
int
>
(
layer_params
,
"stride"
,
1
);
int
filters
=
getParam
<
int
>
(
layer_params
,
"filters"
,
-
1
);
int
filters
=
getParam
<
int
>
(
layer_params
,
"filters"
,
-
1
);
bool
batch_normalize
=
getParam
<
int
>
(
layer_params
,
"batch_normalize"
,
0
)
==
1
;
bool
batch_normalize
=
getParam
<
int
>
(
layer_params
,
"batch_normalize"
,
0
)
==
1
;
...
@@ -563,13 +564,13 @@ namespace cv {
...
@@ -563,13 +564,13 @@ namespace cv {
if
(
flipped
==
1
)
if
(
flipped
==
1
)
CV_Error
(
cv
::
Error
::
StsNotImplemented
,
"Transpose the convolutional weights is not implemented"
);
CV_Error
(
cv
::
Error
::
StsNotImplemented
,
"Transpose the convolutional weights is not implemented"
);
// correct the strange value of pad=1 for kernel_size=1 in the Darknet cfg-file
if
(
pad
)
if
(
kernel_size
<
3
)
pad
=
0
;
padding
=
kernel_size
/
2
;
CV_Assert
(
kernel_size
>
0
&&
filters
>
0
);
CV_Assert
(
kernel_size
>
0
&&
filters
>
0
);
CV_Assert
(
current_channels
>
0
);
CV_Assert
(
current_channels
>
0
);
setParams
.
setConvolution
(
kernel_size
,
pad
,
stride
,
filters
,
current_channels
,
setParams
.
setConvolution
(
kernel_size
,
pad
ding
,
stride
,
filters
,
current_channels
,
batch_normalize
);
batch_normalize
);
current_channels
=
filters
;
current_channels
=
filters
;
...
...
modules/dnn/test/test_darknet_importer.cpp
View file @
f29bc220
...
@@ -106,7 +106,7 @@ public:
...
@@ -106,7 +106,7 @@ public:
std
::
string
cfg
=
findDataFile
(
"dnn/darknet/"
+
name
+
".cfg"
);
std
::
string
cfg
=
findDataFile
(
"dnn/darknet/"
+
name
+
".cfg"
);
std
::
string
model
=
""
;
std
::
string
model
=
""
;
if
(
hasWeights
)
if
(
hasWeights
)
model
=
findDataFile
(
"dnn/darknet/"
+
name
+
".weights"
,
false
);
model
=
findDataFile
(
"dnn/darknet/"
+
name
+
".weights"
);
checkBackend
(
&
inp
,
&
ref
);
checkBackend
(
&
inp
,
&
ref
);
...
@@ -528,6 +528,11 @@ TEST_P(Test_Darknet_layers, reorg)
...
@@ -528,6 +528,11 @@ TEST_P(Test_Darknet_layers, reorg)
testDarknetLayer
(
"reorg"
);
testDarknetLayer
(
"reorg"
);
}
}
TEST_P
(
Test_Darknet_layers
,
convolutional
)
{
testDarknetLayer
(
"convolutional"
,
true
);
}
INSTANTIATE_TEST_CASE_P
(
/**/
,
Test_Darknet_layers
,
dnnBackendsAndTargets
());
INSTANTIATE_TEST_CASE_P
(
/**/
,
Test_Darknet_layers
,
dnnBackendsAndTargets
());
}}
// namespace
}}
// namespace
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