/AIMLProject

Primary LanguagePython

Reconet

Training the model

  • Run train.py with the command line arguments as specified

Training the model with mixed precision:

  • Run amp.py with the command line arguments

Running inference:

  • On a single image:
    • Run style_frame.py
  • On a video:
    • Run style_video.py
  • On live camera input:
    • Run style_camera.py
  • On live camera input (with mixed precision):
    • Run style_camera_fast.py