Spaces:
Sleeping
Sleeping
Update model.py
Browse files
model.py
CHANGED
@@ -13,7 +13,7 @@ List is structured by "B" indicating a residual block followed by the number of
|
|
13 |
"S" is for scale prediction block and computing the yolo loss
|
14 |
"U" is for upsampling the feature map and concatenating with a previous layer
|
15 |
"""
|
16 |
-
|
17 |
(32, 3, 1),
|
18 |
(64, 3, 2),
|
19 |
["B", 1],
|
@@ -130,7 +130,7 @@ class YOLOv3(LightningModule):
|
|
130 |
layers = nn.ModuleList()
|
131 |
in_channels = self.in_channels
|
132 |
|
133 |
-
for module in
|
134 |
if isinstance(module, tuple):
|
135 |
out_channels, kernel_size, stride = module
|
136 |
layers.append(
|
|
|
13 |
"S" is for scale prediction block and computing the yolo loss
|
14 |
"U" is for upsampling the feature map and concatenating with a previous layer
|
15 |
"""
|
16 |
+
config_1 = [
|
17 |
(32, 3, 1),
|
18 |
(64, 3, 2),
|
19 |
["B", 1],
|
|
|
130 |
layers = nn.ModuleList()
|
131 |
in_channels = self.in_channels
|
132 |
|
133 |
+
for module in config_1:
|
134 |
if isinstance(module, tuple):
|
135 |
out_channels, kernel_size, stride = module
|
136 |
layers.append(
|