fangyuanmao / PID

Code for PID: Physics-Informed Diffusion Model for Infrared Image Generation
MIT License
57 stars 5 forks source link

test model PID checkpoint for FLIR, k1=k2=50 #8

Open mzalaki00 opened 3 weeks ago

mzalaki00 commented 3 weeks ago

Hi thanks for your work. can you clearly explain how to use PID checkpoint( for FLIR) for inference with rgb images?? your test script, yaml files belongs to tevnet, kaist and ... ! does it make any difference?

fangyuanmao commented 2 weeks ago

Thanks for your question. We only release the evaluation script for KAIST dataset. You can evaluate the result of KAIST dataset with our introduction. We will release the complete code as soon as the paper is accepted.

mzalaki00 commented 2 weeks ago

sorry but I don't get it, is evaluation script just for KAIST dataset!? why? you mean if i use this model for predicting other dataset, it give wrong output. am i correct?

fangyuanmao commented 2 weeks ago

Thanks for your question. We only release the evaluation script for KAIST dataset. You can evaluate the result of KAIST dataset with our introduction. We will release the complete code as soon as the paper is accepted.

1、"release the evaluation script for KAIST dataset" 2、The architecture of PID is the same one. But checkpoint differs among datasets. We will release the complete version after the paper is accepted.