Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

rename class autoShape -> AutoShape (#3173)

Browse Source 
* rename class autoShape -> AutoShape

follow other class naming convention

* rename class autoShape -> AutoShape

follow other classes' naming convention

* rename class autoShape -> AutoShape
This commit is contained in:
Yonghye Kwon authored and GitHub committed 2021-05-16 15:46:45 +02:00
1 parent 17b0f71538
commit be86c21c73
2 files changed
+6 -6
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

models/yolo.py
+3 -3
View File
@@ -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