gasharper / PyramidFlow

[CVPR 2023] PyramidFlow: High-Resolution Defect Contrastive Localization using Pyramid Normalizing Flow
MIT License
56 stars 14 forks source link

How to inference #4

Closed xcaizewu closed 1 year ago

xcaizewu commented 1 year ago

Thank you for opening the source code. Can you provide the code of model inference? My email is caizewu1994@163.com

gasharper commented 1 year ago

The inference process is the same as the evaluation process. You can refer to the following code in trian.py: https://github.com/gasharper/PyramidFlow/blob/c463b1cb0c2b084cdbae290234e3f8657298e981/train.py#L59-L80

pribadihcr commented 8 months ago

I think we need the inference demo, start from how to load the trained model and infer the test image.