Spaces:
Runtime error
Runtime error
Update hybridnets_test.py
Browse files- hybridnets_test.py +2 -1
hybridnets_test.py
CHANGED
@@ -160,7 +160,8 @@ with torch.no_grad():
|
|
160 |
if imwrite:
|
161 |
cv2.imwrite(f'{output}/{i}.jpg', cv2.cvtColor(ori_imgs[i], cv2.COLOR_RGB2BGR))
|
162 |
|
163 |
-
|
|
|
164 |
print('running speed test...')
|
165 |
with torch.no_grad():
|
166 |
print('test1: model inferring and postprocessing')
|
|
|
160 |
if imwrite:
|
161 |
cv2.imwrite(f'{output}/{i}.jpg', cv2.cvtColor(ori_imgs[i], cv2.COLOR_RGB2BGR))
|
162 |
|
163 |
+
exit()
|
164 |
+
|
165 |
print('running speed test...')
|
166 |
with torch.no_grad():
|
167 |
print('test1: model inferring and postprocessing')
|