ced/update_inference_and_requirements

TimoStoff · Sep 29, 2020 · 1e4fd4d · 1e4fd4d
1 parent c0dfc8e
commit 1e4fd4d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/inference.py b/inference.py
@@ -29,6 +29,8 @@ def legacy_compatibility(args, checkpoint):
     elif args.firenet_legacy:
         args.legacy_norm = True
         final_activation = ''
+    else:
+        return args, checkpoint
     # Make compatible with Henri saved models
     if not isinstance(checkpoint.get('config', None), ConfigParser) or args.e2vid or args.firenet_legacy:
         checkpoint = make_henri_compatible(checkpoint, final_activation)

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,7 @@
+torch
+torchvision
+matplotlib
+opencv-python
+tqdm
+pandas
+h5py