charles/monoloco
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed for tests

Browse Source
This commit is contained in:
Charles Joseph Pierre Beauville 2021-06-28 01:08:34 +02:00
parent f20e17709c
commit 36bdc6335f
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
monoloco/network/net.py
View File

@ -96,7 +96,6 @@ class Loco:
self.model.load_state_dict(torch.load(model_path, map_location=lambda storage, loc: storage)) self.model.load_state_dict(torch.load(model_path, map_location=lambda storage, loc: storage))
if casr: if casr:
print("WTF")
self.turning_model.load_state_dict(torch.load(turning_model_path, self.turning_model.load_state_dict(torch.load(turning_model_path,
map_location=lambda storage, loc: storage)) map_location=lambda storage, loc: storage))
else: else:

3
monoloco/predict.py
View File

@ -132,7 +132,7 @@ def factory_from_args(args):
if args.casr_std: if args.casr_std:
args.casr = 'std' args.casr = 'std'
else: elif args.casr:
args.casr = 'nonstd' args.casr = 'nonstd'
# Patch for stereo images with batch_size = 2 # Patch for stereo images with batch_size = 2
@ -161,7 +161,6 @@ def predict(args):
# Load Models # Load Models
if args.mode in ('mono', 'stereo'): if args.mode in ('mono', 'stereo'):
print(args.casr)
net = Loco( net = Loco(
model=dic_models[args.mode], model=dic_models[args.mode],
mode=args.mode, mode=args.mode,