search

byungjae89 / SPADE-pytorch

PyTorch implementation of "Sub-Image Anomaly Detection with Deep Pyramid Correspondences"
Apache License 2.0
235 stars 43 forks source link

Could we use it with our own dataset? #2

Closed heidarinejad closed 3 years ago

byungjae89 commented 3 years ago

Of course, you could. Add your dataset loader code in src/dataset/your_data.py, and then change dataloader in src/main.py code https://github.com/byungjae89/SPADE-pytorch/blob/master/src/main.py#L58-L63