yolov5/models
Glenn Jocher bfacfc6b4a
Update cls bias init (#5520)
* Update cls bias init

Increased numerical precision. Returns 1.0 probability for single-class datasets now. Addresses https://github.com/ultralytics/yolov5/issues/5357

```python
torch.sigmoid(torch.tensor([math.log(0.6 / (1 - 0.99999))]))
Out[19]: tensor([1.0000])
```

* Update yolo.py
2021-11-05 13:18:46 +01:00
..
hub Add pre-commit CI actions (#4982) 2021-10-28 18:35:01 +02:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py precommit: isort (#5493) 2021-11-04 17:24:25 +01:00
experimental.py precommit: isort (#5493) 2021-11-04 17:24:25 +01:00
tf.py precommit: isort (#5493) 2021-11-04 17:24:25 +01:00
yolo.py Update cls bias init (#5520) 2021-11-05 13:18:46 +01:00
yolov5l.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5m.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5n.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5s.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5x.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00