From 93df56d88da5e2390e0d542ed37a57d48f3f3105 Mon Sep 17 00:00:00 2001 From: "jules@lens" Date: Sat, 5 May 2018 19:32:25 +0200 Subject: new scripts --- Codes/evaluate.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) (limited to 'Codes/evaluate.py') 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): @@ -199,6 +201,15 @@ class GroundTruthLoader(object): gt.append(video_gt) 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 @@ -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) -- cgit v1.2.3-70-g09d2