Reimplementation of Efficient GAN based anomaly detection network by using PyTorch
Efficient GAN based anomaly detection : https://arxiv.org/abs/1802.06222
CONFIG files have to be written in the following format
train_csv_file: ./csv/train.csv
input_size: 64
channel: 1
# the hyper parameters of EGBAD
z_dim: 20
nef: 128
ngf: 128
ndf: 64
batch_size: 10
num_epochs: 500
num_fakeimg: 10
# for wandb
name: first
# training output
save_dir: ./weights
- the network was checked to work properly in only the following image-size
channel : 1, 3
height, width : 64