在跑評價代碼時,遇到no Float Tensor問題的解決辦法(用PIL批量圖片改格式)(未解決)

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'

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章