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-08-06 12:32:37 +08:00
parent dcbc2f49eb
commit 15a465a6d1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@ -58,9 +58,11 @@ def main(only_spec=False, only_color=False):
if only_spec: if only_spec:
# 光谱识别 # 光谱识别
mask_spec = spec_detector.predict(img_data) mask_spec = spec_detector.predict(img_data)
mask_rgb = rgb_detector.predict(rgb_data)
mask_rgb = np.zeros_like(mask_spec, dtype=np.uint8) mask_rgb = np.zeros_like(mask_spec, dtype=np.uint8)
elif only_color: elif only_color:
# rgb识别 # rgb识别
mask_spec = spec_detector.predict(img_data)
mask_rgb = rgb_detector.predict(rgb_data) mask_rgb = rgb_detector.predict(rgb_data)
mask_spec = np.zeros_like(mask_rgb, dtype=np.uint8) mask_spec = np.zeros_like(mask_rgb, dtype=np.uint8)
else: else: