|
|
@@ -296,7 +296,7 @@ class DetectMultiBackend(nn.Module): |
|
|
|
super().__init__() |
|
|
|
w = str(weights[0] if isinstance(weights, list) else weights) |
|
|
|
pt, jit, onnx, xml, engine, coreml, saved_model, pb, tflite, edgetpu, tfjs = self.model_type(w) # get backend |
|
|
|
stride, names = 64, [f'class{i}' for i in range(1000)] # assign defaults |
|
|
|
stride, names = 32, [f'class{i}' for i in range(1000)] # assign defaults |
|
|
|
w = attempt_download(w) # download if not local |
|
|
|
fp16 &= (pt or jit or onnx or engine) and device.type != 'cpu' # FP16 |
|
|
|
if data: # data.yaml path (optional) |