DIRE icon indicating copy to clipboard operation
DIRE copied to clipboard

Easy to Use Request

Open Konerusudhir opened this issue 10 months ago • 6 comments

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.

Konerusudhir avatar Sep 01 '23 01:09 Konerusudhir