白鹭先生 commited on
Commit
280405e
·
1 Parent(s): ac36657
Files changed (2) hide show
  1. gis.py +3 -1
  2. requirements.txt +29 -27
gis.py CHANGED
@@ -2,7 +2,7 @@
2
  Author: Egrt
3
  Date: 2022-03-19 10:25:50
4
  LastEditors: Egrt
5
- LastEditTime: 2022-03-20 14:58:13
6
  FilePath: \Luuu\gis.py
7
  '''
8
  import os
@@ -225,6 +225,8 @@ class GIS(object):
225
  # 如果超出切片预期的大小则关闭切片处理
226
  if image.shape[0] < patch_size or image.shape[1] < patch_size:
227
  self.config['eval_params']['patch_size'] = None
 
 
228
  if 3 < image.shape[2]:
229
  print_utils.print_info(f"Image {in_filepath} has more than 3 channels. Keeping the first 3 channels and discarding the rest...")
230
  image = image[:, :, :3]
 
2
  Author: Egrt
3
  Date: 2022-03-19 10:25:50
4
  LastEditors: Egrt
5
+ LastEditTime: 2022-03-21 00:01:10
6
  FilePath: \Luuu\gis.py
7
  '''
8
  import os
 
225
  # 如果超出切片预期的大小则关闭切片处理
226
  if image.shape[0] < patch_size or image.shape[1] < patch_size:
227
  self.config['eval_params']['patch_size'] = None
228
+ else:
229
+ self.config['eval_params']['patch_size'] = 1024
230
  if 3 < image.shape[2]:
231
  print_utils.print_info(f"Image {in_filepath} has more than 3 channels. Keeping the first 3 channels and discarding the rest...")
232
  image = image[:, :, :3]
requirements.txt CHANGED
@@ -1,27 +1,29 @@
1
- cython
2
- huggingface_hub
3
- scipy==1.4.1
4
- numpy==1.22.3
5
- matplotlib==3.3.2
6
- opencv_python==4.5.4.60
7
- torch==1.7.0
8
- torchvision==0.8.0
9
- tqdm==4.63.0
10
- Pillow==8.2.0
11
- h5py==2.10.0
12
- gradio==2.5.3
13
- jsmin==3.0.1
14
- kornia==0.5.0
15
- shapely==1.8.1
16
- skan==0.10.0
17
- descartes==1.1.0
18
- multiprocess==0.70.12.2
19
- dill==0.3.4
20
- gdal==2.4.4
21
- rasterio
22
- overpy==0.6
23
- pyproj==2.5.0
24
- fiona==1.8.21
25
- zipp==3.4.0
26
- pycocotools==2.0.3
27
- sklearn
 
 
 
1
+ # cython
2
+ # huggingface_hub
3
+ # scipy==1.4.1
4
+ # numpy==1.22.3
5
+ # matplotlib==3.3.2
6
+ # opencv_python==4.5.4.60
7
+ # torch==1.7.0
8
+ # torchvision==0.8.0
9
+ # tqdm==4.63.0
10
+ # Pillow==8.2.0
11
+ # h5py==2.10.0
12
+ # gradio==2.5.3
13
+ # jsmin==3.0.1
14
+ # kornia==0.5.0
15
+ # shapely==1.8.1
16
+ # skan==0.10.0
17
+ # descartes==1.1.0
18
+ # multiprocess==0.70.12.2
19
+ # dill==0.3.4
20
+ # gdal==2.4.4
21
+ # rasterio
22
+ # overpy==0.6
23
+ # pyproj==2.5.0
24
+ # fiona==1.8.21
25
+ # zipp==3.4.0
26
+ # pycocotools==2.0.3
27
+ # sklearn
28
+ rasterio==1.1.3
29
+ gdal==3.0.4