josedolot commited on
Commit
5b5fe0c
·
1 Parent(s): 2698e40

Update hybridnets_test.py

Browse files
Files changed (1) hide show
  1. 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
- # exit()
 
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')