mirrors/yolov7
1
0
Fork 0
mirror of https://github.com/WongKinYiu/yolov7.git synced 2025-06-03 21:54:57 +08:00
Code Issues Projects Releases Wiki Activity

u7 code

Browse Source 
update seg code
This commit is contained in:
Kin-Yiu, Wong 2022-09-02 12:21:29 +08:00 committed by GitHub
parent a4df031e45
commit bcc3eff9e2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
seg/models/common.py
View File

@ -405,7 +405,7 @@ class DetectMultiBackend(nn.Module):
names = model.module.names if hasattr(model, 'module') else model.names # get class names names = model.module.names if hasattr(model, 'module') else model.names # get class names
model.half() if fp16 else model.float() model.half() if fp16 else model.float()
self.model = model # explicitly assign for to(), cpu(), cuda(), half() self.model = model # explicitly assign for to(), cpu(), cuda(), half()
segmentation_model = type(model.model[-1]).__name__ == 'Segment' segmentation_model = type(model.model[-1]).__name__ in ['Segment', 'ISegment', 'IRSegment']
elif jit: # TorchScript elif jit: # TorchScript
LOGGER.info(f'Loading {w} for TorchScript inference...') LOGGER.info(f'Loading {w} for TorchScript inference...')
extra_files = {'config.txt': ''} # model metadata extra_files = {'config.txt': ''} # model metadata