search

proteus1991 / GridDehazeNet

This repo contains the official training and testing codes for our paper: GridDehazeNet: Attention-Based Multi-Scale Network for Image Dehazing.
https://jhc.sjtu.edu.cn/~xiaohongliu/
243 stars 54 forks source link

How to use the own image data? #21

Closed 0119tagawa closed 2 years ago

0119tagawa commented 3 years ago

I was impressive to see your great coding and tremendous results. My question is how to use the our own image data to this model ? I couldn't use them when the data are stored in the ./data/train/indoor/ and ./data/train/outdoor/.

proteus1991 commented 3 years ago

Thanks for your interest. You can customize the test.py to run your own images on this model. More specifically, you can redirect to the image folder that stores those images.