colin
/
Witllm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update show.

This commit is contained in:
Colin 2024-09-16 18:46:09 +08:00
parent 8583bc56d7
commit ce64a2f7aa
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/show.py
View File

@ -6,6 +6,7 @@ import cv2
import math import math
import numpy as np import numpy as np
import os import os
from pathlib import Path
def DumpTensorToImage(tensor, name, forceSquare=False, scale=1.0, Contrast=None, GridValue=None): def DumpTensorToImage(tensor, name, forceSquare=False, scale=1.0, Contrast=None, GridValue=None):
@ -68,6 +69,9 @@ def DumpTensorToImage(tensor, name, forceSquare=False, scale=1.0, Contrast=None,
img[img > 255] = 255 img[img > 255] = 255
imgs = img.astype(np.uint8) imgs = img.astype(np.uint8)
imgs = cv2.applyColorMap(imgs, cv2.COLORMAP_JET) imgs = cv2.applyColorMap(imgs, cv2.COLORMAP_JET)
directory = Path(name).parent
if not directory.is_dir():
directory.mkdir(parents=True, exist_ok=True)
cv2.imwrite(name, imgs) cv2.imwrite(name, imgs)