Browse Source

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

```python
        self.n = len(self.pred)  # number of images (batch size)
```
5.0
Glenn Jocher GitHub 3 years ago
parent
commit
005d7a8c54
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      models/common.py

+ 1
- 1
models/common.py 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


Loading…
Cancel
Save