PIAFusion_pytorch
PIAFusion_pytorch copied to clipboard

I have an Apple Silicon device (MacBook Pro M1Pro). When trying to run `python train_fusion_model.py --dataset_path 'datasets/msrs_train' --epochs 30 --cls_pretrained 'pretrained/best_cls.pth' --batch_size 128 --lr 0.001 --loss_weight '[3, 7, 50]' --save_path...
Is the object detection module including in this code?
thank you for your implementation with pytorch. Can you please explain how to test this model. "python test_fusion_model.py --h5_path 'test_data/MSRS' --save_path 'results/fusion' --fusion_pretrained 'pretrained/fusion_model_epoch_29.pth' " what is the exact path...
Hi, Sorry to have another question for your implementation. In your Dataloader implementation, I notice that you alway return the whole images and feed them to the model for model...
Hi, It is a great work, and thanks for releasing a PyTorch version of your work. I have a quick question for the illumination-aware fusion network training for my own...