duytrq commited on
Commit
09b4579
1 Parent(s): f34343b

turn off half inference mode for consistency

Browse files
Files changed (1) hide show
  1. stamp_processing/detector.py +1 -1
stamp_processing/detector.py CHANGED
@@ -68,7 +68,7 @@ class StampDetector:
68
  for origin_images in batches:
69
  images = [letterbox(x, 640, stride=32)[0] for x in origin_images] # type: ignore
70
  images = list(map(self.process_func_, images))
71
- tensor = torch.stack(images).half()
72
 
73
  with torch.no_grad():
74
  pred = self.model(tensor)[0]
 
68
  for origin_images in batches:
69
  images = [letterbox(x, 640, stride=32)[0] for x in origin_images] # type: ignore
70
  images = list(map(self.process_func_, images))
71
+ tensor = torch.stack(images)#.half()
72
 
73
  with torch.no_grad():
74
  pred = self.model(tensor)[0]