a b/yolov5/utils/__init__.py 

  1
  
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license





  
  

  2
  
"""





  
  

  3
  
utils/initialization





  
  

  4
  
"""





  
  

  5
  





  
  

  6
  





  
  

  7
  
def notebook_init():





  
  

  8
  
    # For YOLOv5 notebooks





  
  

  9
  
    print('Checking setup...')





  
  

  10
  
    from IPython import display  # to display images and clear console output





  
  

  11
  





  
  

  12
  
    from utils.general import emojis





  
  

  13
  
    from utils.torch_utils import select_device  # YOLOv5 imports





  
  

  14
  





  
  

  15
  
    display.clear_output()





  
  

  16
  
    select_device(newline=False)





  
  

  17
  
    print(emojis('Setup complete ✅'))





  
  

  18
  
    return display