提交 a40f615c authored 作者: Glenn Jocher's avatar Glenn Jocher

.half() bug fix

上级 04bdbe41
...@@ -24,7 +24,7 @@ def detect(save_img=False): ...@@ -24,7 +24,7 @@ def detect(save_img=False):
model = torch.load(weights, map_location=device)['model'].float().eval() # load FP32 model model = torch.load(weights, map_location=device)['model'].float().eval() # load FP32 model
imgsz = check_img_size(imgsz, s=model.stride.max()) # check img_size imgsz = check_img_size(imgsz, s=model.stride.max()) # check img_size
if half: if half:
model.float() # to FP16 model.half() # to FP16
# Second-stage classifier # Second-stage classifier
classify = False classify = False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论