From be86c21c73171d954a9ffe6c2d3ffff1b4e04afe Mon Sep 17 00:00:00 2001 From: Yonghye Kwon Date: Sun, 16 May 2021 22:46:45 +0900 Subject: [PATCH] rename class autoShape -> AutoShape (#3173) * rename class autoShape -> AutoShape follow other class naming convention * rename class autoShape -> AutoShape follow other classes' naming convention * rename class autoShape -> AutoShape --- models/common.py | 6 +++--- models/yolo.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/models/common.py b/models/common.py index 9764d4c..689aa0f 100644 --- a/models/common.py +++ b/models/common.py @@ -223,18 +223,18 @@ class NMS(nn.Module): return non_max_suppression(x[0], conf_thres=self.conf, iou_thres=self.iou, classes=self.classes) -class autoShape(nn.Module): +class AutoShape(nn.Module): # input-robust model wrapper for passing cv2/np/PIL/torch inputs. Includes preprocessing, inference and NMS conf = 0.25 # NMS confidence threshold iou = 0.45 # NMS IoU threshold classes = None # (optional list) filter by class def __init__(self, model): - super(autoShape, self).__init__() + super(AutoShape, self).__init__() self.model = model.eval() def autoshape(self): - print('autoShape already enabled, skipping... ') # model already converted to model.autoshape() + print('AutoShape already enabled, skipping... ') # model already converted to model.autoshape() return self @torch.no_grad() diff --git a/models/yolo.py b/models/yolo.py index 314fd80..06b8003 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -215,9 +215,9 @@ class Model(nn.Module): self.model = self.model[:-1] # remove return self - def autoshape(self): # add autoShape module - logger.info('Adding autoShape... ') - m = autoShape(self) # wrap model + def autoshape(self): # add AutoShape module + logger.info('Adding AutoShape... ') + m = AutoShape(self) # wrap model copy_attr(m, self, include=('yaml', 'nc', 'hyp', 'names', 'stride'), exclude=()) # copy attributes return m