diff --git a/README.md b/README.md index bece76e4..019c12bf 100644 --- a/README.md +++ b/README.md @@ -136,7 +136,7 @@ If our work is useful for your research, please consider citing: ### License -This project is licensed under S-Lab License 1.0. Redistribution and use for non-commercial purposes should follow this license. +This project is licensed under NTU S-Lab License 1.0. Redistribution and use should follow this license. ### Acknowledgement diff --git a/basicsr/archs/vqgan_arch.py b/basicsr/archs/vqgan_arch.py index f6dfcf4c..5ac69263 100644 --- a/basicsr/archs/vqgan_arch.py +++ b/basicsr/archs/vqgan_arch.py @@ -41,10 +41,10 @@ def forward(self, z): mean_distance = torch.mean(d) # find closest encodings - # min_encoding_indices = torch.argmin(d, dim=1).unsqueeze(1) - min_encoding_scores, min_encoding_indices = torch.topk(d, 1, dim=1, largest=False) + min_encoding_indices = torch.argmin(d, dim=1).unsqueeze(1) + # min_encoding_scores, min_encoding_indices = torch.topk(d, 1, dim=1, largest=False) # [0-1], higher score, higher confidence - min_encoding_scores = torch.exp(-min_encoding_scores/10) + # min_encoding_scores = torch.exp(-min_encoding_scores/10) min_encodings = torch.zeros(min_encoding_indices.shape[0], self.codebook_size).to(z) min_encodings.scatter_(1, min_encoding_indices, 1) @@ -66,7 +66,6 @@ def forward(self, z): "perplexity": perplexity, "min_encodings": min_encodings, "min_encoding_indices": min_encoding_indices, - "min_encoding_scores": min_encoding_scores, "mean_distance": mean_distance } diff --git a/facelib/detection/yolov5face/face_detector.py b/facelib/detection/yolov5face/face_detector.py index 0103411e..79fdba0c 100644 --- a/facelib/detection/yolov5face/face_detector.py +++ b/facelib/detection/yolov5face/face_detector.py @@ -17,7 +17,7 @@ scale_coords_landmarks, ) -IS_HIGH_VERSION = tuple(map(int, torch.__version__.split('+')[0].split('.')[:3])) >= (1, 9, 0) +IS_HIGH_VERSION = tuple(map(int, torch.__version__.split('+')[0].split('.')[:2])) >= (1, 9, 0) def isListempty(inList):