|
|
@@ -62,7 +62,7 @@ if __name__ == '__main__': |
|
|
if opt.include_nms: |
|
|
if opt.include_nms: |
|
|
model.model[-1].include_nms = True |
|
|
model.model[-1].include_nms = True |
|
|
y = None |
|
|
y = None |
|
|
# TorchScript export 1 |
|
|
|
|
|
|
|
|
# TorchScript export 11 |
|
|
try: |
|
|
try: |
|
|
print('\nStarting TorchScript export with torch %s...' % torch.__version__) |
|
|
print('\nStarting TorchScript export with torch %s...' % torch.__version__) |
|
|
f = opt.weights.replace('.pt', '.torchscript.pt') # filename |
|
|
f = opt.weights.replace('.pt', '.torchscript.pt') # filename |
|
|
|