diff --git a/util.py b/util.py index 033be7f3c2552bcb9d2b84ab855f2bad22536d2c..580841da0f854682fee6b65c35a09e1867388064 100644 --- a/util.py +++ b/util.py @@ -291,7 +291,7 @@ def generate_test_vid_names(data_dict, dset): '''--generates test folder names''' - if dset == 'Thermal_Fall' or 'Thermal_Dummy': + if dset == 'Thermal_Fall' or dset == 'Thermal_Dummy': vid_base_name = 'Fall' elif dset == 'Thermal_Intrusion': vid_base_name = 'Intru'