We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent da693f8 commit 57909c1Copy full SHA for 57909c1
demos/common/python/models/yolo.py
@@ -245,10 +245,9 @@ def _get_output_info(self):
245
246
output_info = {}
247
for i, (name, layer) in enumerate(outputs):
248
- if name.startswith('conv'):
249
- shape = layer.shape
250
- yolo_params = self.Params(num, shape[2:4], anchors, masks[len(output_info)])
251
- output_info[name] = (shape, yolo_params)
+ shape = layer.shape
+ yolo_params = self.Params(num, shape[2:4], anchors, masks[len(output_info)])
+ output_info[name] = (shape, yolo_params)
252
return output_info
253
254
@staticmethod
0 commit comments