diff options
| author | jules@lens <julescarbon@gmail.com> | 2018-05-05 19:32:25 +0200 |
|---|---|---|
| committer | jules@lens <julescarbon@gmail.com> | 2018-05-05 19:32:25 +0200 |
| commit | 93df56d88da5e2390e0d542ed37a57d48f3f3105 (patch) | |
| tree | 9144319dbe9fb2f70f03fea0c68cf043db38c585 /Codes/evaluate.py | |
| parent | 91e5f1ffb152e1b729fe9d530d9f01e73017abbf (diff) | |
Diffstat (limited to 'Codes/evaluate.py')
| -rw-r--r-- | Codes/evaluate.py | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/Codes/evaluate.py b/Codes/evaluate.py index cd0fcef..4909e50 100644 --- a/Codes/evaluate.py +++ b/Codes/evaluate.py @@ -114,8 +114,10 @@ class GroundTruthLoader(object): gt = self.__load_shanghaitech_gt() elif dataset == GroundTruthLoader.TOY_DATA: gt = self.__load_toydata_gt() + #else: + # gt = self.__load_ucsd_avenue_subway_gt(dataset) else: - gt = self.__load_ucsd_avenue_subway_gt(dataset) + gt = self.__load_no_anomaly_gt(dataset) return gt def __load_ucsd_avenue_subway_gt(self, dataset): @@ -200,6 +202,15 @@ class GroundTruthLoader(object): return gt @staticmethod + def __load_no_anomaly_gt(): + gt = [] + for i in xrange(27): + length = 64 + video_gt = np.zeros((length,), dtype=np.uint8) + gt.append(video_gt) + return gt + + @staticmethod def get_pixel_masks_file_list(dataset): # pixel mask folder pixel_mask_folder = os.path.join(DATA_DIR, dataset, 'pixel_masks') @@ -573,4 +584,4 @@ if __name__ == '__main__': if eval_type == 'test_func': test_func() else: - evaluate(eval_type, file_path)
\ No newline at end of file + evaluate(eval_type, file_path) |