numpy
matplotlib
h5py
tqdm
rich
torch
torchvision
pyecharts
