image_t = transform(image).to(device)

mudit-dhawan / GAME-ON

2 stars 3 forks source link

image_t = transform(image).to(device) #2

Open xiaozhi1233 opened 7 months ago

xiaozhi1233 commented 7 months ago

想问下imageGraph.py这个文件里的： image_t = transform(image).to(device) image_t = image_t.unsqueeze(0) # add a batch dimension image_t = image_t[:,:3, :, :] 这里的transform是定义的函数还是类？为什么没有其具体代码，运行会报错

mudit-dhawan commented 7 months ago

there was an error, please change it to image_transform and add that to the argument list as well.