upload ssd ppt

304886938 · Jul 19, 2020 · e7c7bda · e7c7bda
1 parent 51e0270
commit e7c7bda
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 1 deletion.
diff --git a/course_ppt/SSD.pdf b/course_ppt/SSD.pdf
diff --git a/pytorch_object_detection/ssd/README.md b/pytorch_object_detection/ssd/README.md
@@ -42,7 +42,7 @@
 * https://b23.tv/GJnkOD
 
 ## 进一步了解该项目，以及对SSD算法代码的分析可参考我的bilibili
-
+* https://www.bilibili.com/video/BV1vK411H771/
 
 ## Resnet50 + SSD算法框架图
 ![Resnet50 SSD](https://github.com/WZMIAOMIAO/deep-learning-for-image-processing/raw/master/pytorch_object_detection/ssd/res50_ssd.png) 
diff --git a/pytorch_object_detection/ssd/src/ssd_model.py b/pytorch_object_detection/ssd/src/ssd_model.py
@@ -95,6 +95,7 @@ def bbox_view(self, features, loc_extractor, conf_extractor):
         for f, l, c in zip(features, loc_extractor, conf_extractor):
             # [batch, n*4, feat_size, feat_size] -> [batch, 4, -1]
             locs.append(l(f).view(f.size(0), 4, -1))
+            # [batch, n*classes, feat_size, feat_size] -> [batch, classes, -1]
             confs.append(c(f).view(f.size(0), self.num_classes, -1))
 
         locs, confs = torch.cat(locs, 2).contiguous(), torch.cat(confs, 2).contiguous()