Fried-Rice-Lab / FriedRiceLab

Official repository of the Fried Rice Lab, including code resources of the following our works: ESWT [arXiv], etc. This repository also implements many useful features and out-of-the-box image restoration models.
MIT License
209 stars 30 forks source link

Calling the model for inference on one image #20

Closed LilyTheBear closed 1 year ago

LilyTheBear commented 1 year ago

Hi,

From the code, it is not clear how do you call the model for inference on one image, and what data format you use. Would you mind to provide one example.

jnpngshiii commented 1 year ago

Making inferences on a single image is no different from making inferences on multiple images. Just put this image under a folder and treat this folder as a dataset.

As for the format of the image, both jpg and png are fine. You can refer to BasicSR's documentation to determine if more formats are supported.