Sa-m's picture
Update models/yolo.py
2f36488