fix: label tensors used directly in % string formatting crash

[andrewwhitecdw]

Problem

fix: label tensors used directly in % string formatting crash

Fix

Replace:

            id_img, label = data
            id_img_flip = Variable(fliplr(id_img).cuda())
            id_img = Variable(id_img.cuda())
            bg_img, label2 = data2

with:

            id_img, label = data
            label = label.item()
            id_img_flip = Variable(fliplr(id_img).cuda())
            id_img = Variable(id_img.cuda())
            bg_img, label2 = data2
            label2 = label2.item()

Files changed

• visual_tools/test_folder.py