import PIL.Image as Image
import cv2
import sys
import numpy as np
import os
mySys = 'win' # win
filePath1 = '/home/nk/tmp/HKU-ISggg/'
list_png1 = os.listdir(filePath1)
list_png1 = sorted(list_png1)
filePath2 = '/home/nk/tmp/new/'
list_png2 = os.listdir(filePath2)
list_png2 = sorted(list_png2)
if mySys == 'mac':
list_png1 = list_png1[1:]
list_png2 = list_png2[1:]
print('1', len(list_png1))
print('2', len(list_png2))
# ECSSD PASCALS DUTS-TE DUT-OMRON HKU-IS
if __name__ == '__main__':
for i in list_png1:
#print(i)
# print(type(i))
# img = cv2.imread(filePath1 + i)
print(i)
img = Image.open(filePath1 + i).convert('RGB')
#i = i.split('.')[0]
# print(i)
output = '/home/nk/tmp/HKU-ISnew/' + i
print(output)
img.save(output)
#print(type(img))
# img = 255. - img
#i = i.split('.')[0]
#print(i)
#output = 'home/nk/tmp/x/' + i + '.png'
#print(output)
#cv2.imwrite(output, img)
遇到問題如下:
Traceback (most recent call last):
File "main.py", line 59, in <module>
main(config)
File "main.py", line 40, in main
print(thread.run())
File "/home/nk/zjc/PycharmProjects/Nk_val/fun-python/Evaluate-SOD-master/evaluator.py", line 22, in run
s = self.Eval_Smeasure() # todo
File "/home/nk/zjc/PycharmProjects/Nk_val/fun-python/Evaluate-SOD-master/evaluator.py", line 123, in Eval_Smeasure
Q = torch.FLoatTensor([0.0])
AttributeError: module 'torch' has no attribute 'FLoatTensor'