dailingna
dailingna
i have the same question!!!
i have the same question
i have the same question
i have the same question
How can you create CelebA-HQ mask?
我已经找到问题的关键所在了！！！ 1.需要在生成时候将 output_tensor = F.log_softmax(output_tensor[0], dim=1) 修改为 output_tensor = F.softmax(output_tensor[0], dim=1) 。 2.生成出来的output_tensor尺寸是1/4/H/W,其中的output_tensor[0,0,:,:]中的白色区域（概率大的数据）位置表示的是背景，而我最一开始时候的初始化均为'-float(inf)',这将导致最后融合的时候出现问题 细节的处理代码如下： ``` def ClothSegMultiGen(self,img_cv,size=-1,): img = Image.fromarray(cv2.cvtColor(img_cv, cv2.COLOR_BGR2RGB)) w,h = img.size body_boxes, _, sub_bodys = self.detectbody.gen_bodybox(img_cv) print('监测到的人物个数为：',len(sub_bodys))...
在这个代码中 ，这两个问题我好像没遇到过，实际记不太清了。 关于“TypeError: Caught TypeError in DataLoader worker process 0.”这个问题 大概是因为你在dataloader里面你的worker设置的问题，设置成0应该是不报错了 关于“'float' object cannot be interpreted as an integer” 具体报错的定位是哪里呢？ 我的环境是torch1.8 torchvision0.9 python3.6.5 cuda11.1
会出现这个问题。在输入网络中的图片过大时 就会显存不够
图片多大尺寸？