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')
|