当我运行一段代码的时候,报错
Traceback (most recent call last):
File "/home/wang/Desktop/wjl-project/Adversarial_camou-main/Visualize-Shirts.py", line 43, in <module>
tex,tex_trouser = trainer.update_mesh(type='determinate')
File "/home/wang/Desktop/wjl-project/Adversarial_camou-main/train.py", line 354, in update_mesh
gb_tshirt = -(-(self.seeds_tshirt + 1e-20).log() + 1e-20).log()
AttributeError: 'PatchTrainer' object has no attribute 'seeds_tshirt'
下面是我跳转到错误的350行代码的位置,我不知道是什么引起的错误。
def update_mesh(self, tau=0.3, type='gumbel'):
# camouflage:
# print(f'Updating mesh with seeds_tshirt: {self.seeds_tshirt}')
# print(f'Updating mesh with seeds_trouser: {self.seeds_trouser}')
prob_map = prob_fix_color(self.tshirt_point, self.coordinates, self.colors, self.h, self.w, blur=self.args.blur).unsqueeze(0)
prob_trouser = prob_fix_color(self.trouser_point, self.coordinates_t, self.colors, self.h_t, self.w_t, blur=self.args.blur).unsqueeze(0)
prob_map = self.camouflage_kernel(prob_map)
prob_trouser = self.camouflage_kernel(prob_trouser)
prob_map = prob_map.squeeze(0).permute(1, 2, 0)
prob_trouser = prob_trouser.squeeze(0).permute(1, 2, 0)
gb_tshirt = -(-(self.seeds_tshirt + 1e-20).log() + 1e-20).log()
gb_trouser = -(-(self.seeds_trouser + 1e-20).log() + 1e-20).log()
tex = gumbel_color_fix_seed(prob_map, gb_tshirt, self.colors, tau=tau, type=type)
tex_trouser = gumbel_color_fix_seed(prob_trouser, gb_trouser, self.colors, tau=tau, type=type)
tex = self.expand_kernel(self.color_transform(tex.permute(0, 3, 1, 2))).permute(0, 2, 3, 1)
tex_trouser = self.expand_kernel(self.color_transform(tex_trouser.permute(0, 3, 1, 2))).permute(0, 2, 3, 1)
self.mesh_tshirt.textures = TexturesUV(maps=tex, faces_uvs=self.faces, verts_uvs=self.verts_uv)
self.mesh_trouser.textures = TexturesUV(maps=tex_trouser, faces_uvs=self.faces_trouser, verts_uvs=self.verts_uv_trouser)
return tex, tex_trouser