File size: 963 Bytes
8b4f646 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import os
netron_extensions = set([
".armnn",
".caffemodel",
".circle",
".ckpt",
".cmf",
".dlc",
".dnn",
".h5",
".har",
".hd5",
".hdf5",
".hn",
".keras",
".kmodel",
".lite",
".mar",
".meta",
".mge",
".mlmodel",
".mlnet",
".mlpackage",
".mnn",
".model",
".nb",
".ngf",
".nn",
".nnp",
".npy",
".npz",
".om",
".onnx",
".ort",
".paddle",
".param",
".pb",
".pbtxt",
".pdiparams",
".pdmodel",
".pdopt",
".pdparams",
".pkl",
".prototxt",
".pt",
".pth",
".ptl",
".rknn",
".safetensors",
".t7",
".tfl",
".tflite",
".tmfile",
".tm",
".tnnproto",
".torchscript",
".uff",
".xmodel",
])
def is_netron_file(file: str) -> bool:
_, ext = os.path.splitext(file)
if ext is None:
return False
return ext in netron_extensions
def build_netron_iframe(url: str) -> str:
return """
<iframe src="https://netron.app?url={url}" style="width:100%; height:70vh;"></iframe>
""".format(url=url) |