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

[ext] 添加色彩

Browse Source
This commit is contained in:
FEIJINTI 2022-08-04 19:16:03 +08:00
parent 5a34a5a7bd
commit 14878662c1
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
main_test.py
View File

@ -59,11 +59,18 @@ class TestMain:
with open(os.path.join(test_path, spec_file_name), 'rb') as f: with open(os.path.join(test_path, spec_file_name), 'rb') as f:
data = f.read() data = f.read()
spec_img = transmit.BeforeAfterMethods.spec_data_post_process(data) spec_img = transmit.BeforeAfterMethods.spec_data_post_process(data)
if convert:
spec_img_show = np.asarray(np.clip(spec_img[..., [21, 3, 0]] * 255, a_min=0, a_max=255),
dtype=np.uint8)
cv2.imwrite(rgb_file_name + '.bmp', spec_img_show[..., ::-1])
spec_mask = self.test_spec(spec_img, img_name=spec_file_name) spec_mask = self.test_spec(spec_img, img_name=spec_file_name)
if test_rgb: if test_rgb:
with open(os.path.join(test_path, rgb_file_name), 'rb') as f: with open(os.path.join(test_path, rgb_file_name), 'rb') as f:
data = f.read() data = f.read()
rgb_img = transmit.BeforeAfterMethods.rgb_data_post_process(data) rgb_img = transmit.BeforeAfterMethods.rgb_data_post_process(data)
if convert:
cv2.imwrite(os.path.join(test_path, rgb_file_name + '.bmp'), rgb_img[..., ::-1])
rgb_mask = self.test_rgb(rgb_img, img_name=rgb_file_name) rgb_mask = self.test_rgb(rgb_img, img_name=rgb_file_name)
if test_rgb and test_spectra: if test_rgb and test_spectra:
if get_delta: if get_delta:
@ -165,8 +172,8 @@ class TestMain:
if __name__ == '__main__': if __name__ == '__main__':
import argparse import argparse
parser = argparse.ArgumentParser(description='Run image test or ') parser = argparse.ArgumentParser(description='Run image test or ')
tester = TestMain() tester = TestMain()
tester.pony_run(test_path=r'/home/lzy/2022.7.30/tobacco_v1_0/saved_img/', tester.pony_run(test_path=r'E:\zhouchao\8.4\yangeng',
test_rgb=True, test_spectra=True, get_delta=False) test_rgb=True, test_spectra=True, get_delta=False, convert=True)