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(
|