Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Update export.py (#2909)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-04-23 23:50:02 +02:00
1 parent 28db237639
commit a2a514dec8
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
models/export.py
+1 -1
View File
@@ -112,7 +112,7 @@ if __name__ == '__main__':
try: try:
import coremltools as ct import coremltools as ct
print(f'{prefix} starting export with coremltools {onnx.__version__}...') print(f'{prefix} starting export with coremltools {ct.__version__}...')
# convert model from torchscript and apply pixel scaling as per detect.py # convert model from torchscript and apply pixel scaling as per detect.py
model = ct.convert(ts, inputs=[ct.ImageType(name='image', shape=img.shape, scale=1 / 255.0, bias=[0, 0, 0])]) model = ct.convert(ts, inputs=[ct.ImageType(name='image', shape=img.shape, scale=1 / 255.0, bias=[0, 0, 0])])
f = opt.weights.replace('.pt', '.mlmodel') # filename f = opt.weights.replace('.pt', '.mlmodel') # filename