YOLO / yolo /utils /model_utils.py

Commit History

πŸ’š [Fix] MPS CI bugs, enable using MPS
4056352

henry000 commited on

βœ… [Pass] test with fixup choosing device
ecc08c8

henry000 commited on

πŸš‘οΈ [Fix] get_device typo, not return cpu
3b31306

henry000 commited on

πŸ”¨ [Add] Anc2Box conveter, for YOLOv7's output
d5a73bd

henry000 commited on

βœ… [Pass] test with lower system requirements
2ee7407

henry000 commited on

πŸ”€ [Merge] branch 'TRAIN' into TEST
1ded329

henry000 commited on

πŸ› [Fix] #32 bugs, turn 8400->auto anchor size
7f8ebc5

henry000 commited on

✨ [New] local mAP calculation, AP.5:.95 - mAP.5
21a413f

henry000 commited on

πŸ”¨ [Add] the PostProccess class to convert predicts
475302b

henry000 commited on

πŸš€ [New] DDP mode for training model
1fe2937

henry000 commited on

🚧 [WIP] DDP for model training
7d7e199

henry000 commited on

πŸ”§ [Update] all structure of yaml, more readability
97e9dcb

henry000 commited on

🚚 [Rename] All utils, tools, higher readability
dcceddd

henry000 commited on