白鹭先生
commited on
Commit
·
280405e
1
Parent(s):
ac36657
修复
Browse files- gis.py +3 -1
- 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-
|
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
|