NanjingForestryUniversity/supermachine-tobacco
3
0
Fork 0
mirror of https://github.com/NanjingForestryUniversity/supermachine-tobacco.git synced 2025-11-08 14:23:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

改阈值

Browse Source
This commit is contained in:
FEIJINTI 2022-07-27 08:20:58 +08:00
parent 5ace1f858b
commit ea7243da29
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.py
View File

@ -4,7 +4,7 @@ import numpy as np
class Config: class Config:
# 文件相关参数 # 文件相关参数
nRows, nCols, nBands, spec_size_threshold, = 256, 1024, 22, 3 nRows, nCols, nBands, spec_size_threshold = 256, 1024, 22, 3
nRgbRows, nRgbCols, nRgbBands, rgb_size_threshold = 1024, 4096, 3, 4 nRgbRows, nRgbCols, nRgbBands, rgb_size_threshold = 1024, 4096, 3, 4
# 需要设置的谱段等参数 # 需要设置的谱段等参数

2
main.py
View File

@ -27,6 +27,7 @@ def main():
# 读取(开启一个管道) # 读取(开启一个管道)
if len(data) < 3: if len(data) < 3:
threshold = int(float(data)) threshold = int(float(data))
Config.spec_size_threshold = threshold
print("[INFO] Get threshold: ", threshold) print("[INFO] Get threshold: ", threshold)
continue continue
else: else:
@ -34,6 +35,7 @@ def main():
rgb_data = os.read(fd_rgb, total_rgb) rgb_data = os.read(fd_rgb, total_rgb)
if len(rgb_data) < 3: if len(rgb_data) < 3:
rgb_threshold = int(float(rgb_data)) rgb_threshold = int(float(rgb_data))
Config.rgb_size_threshold = rgb_threshold
print(rgb_threshold) print(rgb_threshold)
continue continue
else: else: