turn off half inference mode for consistency
Browse files
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)
|
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]
|