search

PolynomialQian / RD3D

RD3D: RGB-D Salient Object Detection via 3D Convolutional Neural Networks
MIT License
49 stars 8 forks source link

when i want to run tarin.py file , this error show me, how can fix this error?? i should run this project this is important for me #7

Open BenyaminMoadab opened 3 years ago

BenyaminMoadab commented 3 years ago

IN[1]: runfile('C:/Users/Asus/Desktop/RD3D-master/train.py', wdir='C:/Users/Asus/Desktop/RD3D-master')

Namespace(batchsize=10, clip=0.5, data_dir='', epochs=50, hflip=False, lr=0.000125, lr_decay_epochs=[120, 160, 200], lr_decay_rate=0.1, lr_decay_steps=20, lr_scheduler='cosine', momentum=0.9, optim='adamW', output_dir='./output\1630156223', trainsize=352, vflip=False, warmup_epoch=-1, warmup_multiplier=100, weight_decay=0.0001) [08/28 17:40:23 rd3d]: Full config saved to ./output\1630156223\config.json ========>num_gpus:1==========

Traceback (most recent call last):

File "C:\Users\Asus\Desktop\RD3D-master\train.py", line 169, in ckpt_path = main(opt)

File "C:\Users\Asus\Desktop\RD3D-master\train.py", line 77, in main train_loader = build_loader(opt)

File "C:\Users\Asus\Desktop\RD3D-master\train.py", line 60, in build_loader train_loader = get_loader(image_root, gt_root, depth_root, batchsize=opt.batchsize * num_gpus,

File "C:\Users\Asus\Desktop\RD3D-master\data.py", line 127, in get_loader dataset = SalObjDataset(image_root, gt_root, depth_root, trainsize, hflip=hflip, vflip=vflip)

File "C:\Users\Asus\Desktop\RD3D-master\data.py", line 23, in init self.filter_files()

File "C:\Users\Asus\Desktop\RD3D-master\data.py", line 83, in filter_files assert len(self.images) == len(self.gts)

AssertionError