Skip to main content

Browse the archive

https://github.com/samet-akcay/ganomaly
05 April 2024, 20:29:26 UTC
Revision 63be7f45acde2b61ab5f3e3110010472c7b01122 authored by Samet Akcay on 07 August 2019, 15:38:30 UTC, committed by Samet Akcay on 07 August 2019, 15:38:30 UTC 
The following are the modifications made to lib/model.py file.
 
1. Added BaseModel class.
2. Refactored ganomaly class.
3. Updated the feature matching loss of the generator network.
4. Updated the discriminator loss.
1 parent cd5f78e
Raw File
Tip revision: 63be7f45acde2b61ab5f3e3110010472c7b01122 authored by Samet Akcay on 07 August 2019, 15:38:30 UTC
The following are the modifications made to lib/model.py file.
Tip revision: 63be7f4
train.py 
"""
TRAIN GANOMALY

. Example: Run the following command from the terminal.
    run train.py                             \
        --model ganomaly                        \
        --dataset UCSD_Anomaly_Dataset/UCSDped1 \
        --batchsize 32                          \
        --isize 256                         \
        --nz 512                                \
        --ngf 64                               \
        --ndf 64
"""


##
# LIBRARIES
from __future__ import print_function

from options import Options
from lib.data import load_data
from lib.model import Ganomaly

##
# def main():
""" Training
"""

##
# ARGUMENTS
opt = Options().parse()

##
# LOAD DATA
dataloader = load_data(opt)

##
# LOAD MODEL
model = Ganomaly(opt, dataloader)

##
# TRAIN MODEL
model.train()

# if __name__ == '__main__':
#     main()
back to top