Update Detections() self.n comment (#2620)

```python
        self.n = len(self.pred)  # number of images (batch size)
```
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-03-26 21:19:15 +01:00
1 parent 8f6e447729
commit 005d7a8c54
1 file changed
+1 -1
+1 -1
View File
@@ -247,7 +247,7 @@ class Detections:
self.xywh = [xyxy2xywh(x) for x in pred] # xywh pixels
self.xyxyn = [x / g for x, g in zip(self.xyxy, gn)] # xyxy normalized
self.xywhn = [x / g for x, g in zip(self.xywh, gn)] # xywh normalized
self.n = len(self.pred)
self.n = len(self.pred) # number of images (batch size)
self.t = tuple((times[i + 1] - times[i]) * 1000 / self.n for i in range(3)) # timestamps (ms)
self.s = shape # inference BCHW shape