mirrors
/
mmyolo
mirror of https://github.com/open-mmlab/mmyolo.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix gardcam error using in mmdet (#779)

tutorials
Haian Huang(深度眸) 2023-05-17 19:39:43 +08:00 committed by GitHub
parent a3b6ae1d65
commit 1e5372e993
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mmyolo/utils/boxam_utils.py
View File

@ -202,8 +202,10 @@ class BoxAMDetectorWrapper(nn.Module):
if self.is_need_loss:
# Maybe this is a direction that can be optimized
# self.detector.init_weights()
if hasattr(self.detector.bbox_head, 'head_module'):
self.detector.bbox_head.head_module.training = True
else:
self.detector.bbox_head.training = True
if hasattr(self.detector.bbox_head, 'featmap_sizes'):
# Prevent the model algorithm error when calculating loss
self.detector.bbox_head.featmap_sizes = None
@ -219,7 +221,10 @@ class BoxAMDetectorWrapper(nn.Module):
return [loss]
else:
if hasattr(self.detector.bbox_head, 'head_module'):
self.detector.bbox_head.head_module.training = False
else:
self.detector.bbox_head.training = False
with torch.no_grad():
results = self.detector.test_step(self.input_data)
return results