Skip to content
This repository has been archived by the owner on Aug 7, 2023. It is now read-only.

Add CLI params support + MobileNetV2 model + small refactoring #9

Merged
merged 10 commits into from
Mar 8, 2021
Prev Previous commit
Change output variable to more meaningful name
  • Loading branch information
Remigiusz Poltorak committed Feb 6, 2021
commit 645e594b0cde360d28f9cbd4670a047a9fdf333c
8 changes: 4 additions & 4 deletions mask_detector/models/mobile_net_v2.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,10 +21,10 @@ def __init__(self):
Sigmoid()
)

def forward(self, x):
def forward(self, output):
with torch.no_grad():
x = self.features(x)
output = self.features(output)

x = self.classifier(x)
output = self.classifier(output)

return x
return output