mirror of
https://github.com/gaomingqi/Track-Anything.git
synced 2025-12-16 16:37:58 +01:00
13 lines
377 B
Python
13 lines
377 B
Python
|
import torchvision.transforms as transforms
|
||
|
|
|
||
|
|
im_mean = (124, 116, 104)
|
||
|
|
|
||
|
|
im_normalization = transforms.Normalize(
|
||
|
|
mean=[0.485, 0.456, 0.406],
|
||
|
|
std=[0.229, 0.224, 0.225]
|
||
|
|
)
|
||
|
|
|
||
|
|
inv_im_trans = transforms.Normalize(
|
||
|
|
mean=[-0.485/0.229, -0.456/0.224, -0.406/0.225],
|
||
|
|
std=[1/0.229, 1/0.224, 1/0.225])
|