DIRE
DIRE copied to clipboard
Easy to Use Request
Currently, the setup is divided into 2 steps. Step 1: Copute_dire.sh to generate an DIRE image from Input image(Real or Fake) (Distributed computing on N GPUS) Step 2: demo.py to generate probability from DIRE image.
This setup makes it very hard to test and use the model.
I think if we can simplify it to one batched function call to execute both steps will make the model more usable to other researchers or end users.
Example: Create inference.py file which takes a directory as input and generates probabilities in a CSV file on a Single GPU or M2 Mac.