-
Notifications
You must be signed in to change notification settings - Fork 10
/
main.py
26 lines (20 loc) · 788 Bytes
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from configuration.config import parse_arguments_main
from paradigms.centralized.c2d import CentralizedAD2D
from paradigms.centralized.c3d import CentralizedAD3D
from paradigms.federated.f2d import FederatedAD2D
def main(args):
# centralized learning for 2d anomaly detection
if args.paradigm == 'c2d':
work = CentralizedAD2D(args=args)
work.run_work_flow()
# centralized learning for 3d anomaly detection
if args.paradigm == 'c3d':
work = CentralizedAD3D(args=args)
work.run_work_flow()
# federated learning for 2d anomaly detection
if args.paradigm == 'f2d':
work = FederatedAD2D(args=args)
work.run_work_flow()
if __name__ == '__main__':
args = parse_arguments_main()
main(args)