torch==2.4.1+cu118
labelme
numpy
Pillow
timm 
opencv-python 
einops
shapely