chuuhtetnaing commited on
Commit
67569e0
·
1 Parent(s): aeb817d

upgrade numpy

Browse files
Files changed (2) hide show
  1. hrnet_ocr.py +1 -1
  2. requirements.txt +1 -1
hrnet_ocr.py CHANGED
@@ -209,7 +209,7 @@ class HighResolutionNet(nn.Module):
209
  num_blocks=4 * [num_blocks], num_channels=num_channels)
210
 
211
  if self.ocr_on:
212
- last_inp_channels = np.int(np.sum(pre_stage_channels))
213
  ocr_mid_channels = 2 * ocr_width
214
  ocr_key_channels = ocr_width
215
 
 
209
  num_blocks=4 * [num_blocks], num_channels=num_channels)
210
 
211
  if self.ocr_on:
212
+ last_inp_channels = int(np.sum(pre_stage_channels))
213
  ocr_mid_channels = 2 * ocr_width
214
  ocr_key_channels = ocr_width
215
 
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  adamp==0.3.0
2
- numpy==1.21.2
3
  opencv-python==4.12.0.88
4
  opencv-python-headless==4.12.0.88
5
  pytorch-msssim==1.0.0
 
1
  adamp==0.3.0
2
+ numpy==1.24.4
3
  opencv-python==4.12.0.88
4
  opencv-python-headless==4.12.0.88
5
  pytorch-msssim==1.0.0