search

mjkwon2021 / CAT-Net

Official code for CAT-Net: Compression Artifact Tracing Network. Image manipulation detection and localization.
222 stars 26 forks source link

Inference on CPU #21

Open Hamdi-Ben-Abdallah opened 2 years ago

Hamdi-Ben-Abdallah commented 2 years ago

Hello,

I would like to test your code and your pre-trainer model and I wanted to know if it works only on CPU? Do you know the execution time to process an image?

Thanks in advance.

CauchyComplete commented 2 years ago

Hi, I'm sorry to inform you that this code cannot run without CUDA. You'll see so many .cuda() in the whole project. If you wish to run it, you need to manually modify the entire project.