joewybean / MMI-Det

The official version of the paper "MMI-Det: Exploring Multi-Modal Integration for Visible and Infrared Object Detection"
Apache License 2.0
4 stars 0 forks source link

'Model' object has no attribute 'Enhance' #4

Open NhanDoremon opened 7 hours ago

NhanDoremon commented 7 hours ago

I am encountering this problem when running test.py Can you you let me know how to fix it, thank you in advance. Capture

NhanDoremon commented 7 hours ago

Traceback (most recent call last): File "D:\OneDrive - Sejong University\NgheNhan_Master\Research\Facial Disguise Detection\Test Model\MMI-Det\test.py", line 339, in test(opt.data, File "D:\OneDrive - Sejong University\NgheNhan_Master\Research\Facial Disguise Detection\Test Model\MMI-Det\test.py", line 123, in test out_all, Contrastive_Loss = model(img_rgb, img_ir, augment=augment) File "C:\Users\Nghe_Nhan\anaconda3\envs\MMI_Det\lib\site-packages\torch\nn\modules\module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, *kwargs) File "C:\Users\Nghe_Nhan\anaconda3\envs\MMI_Det\lib\site-packages\torch\nn\modules\module.py", line 1747, in _call_impl return forward_call(args, **kwargs) File "D:\OneDrive - Sejong University\NgheNhan_Master\Research\Facial Disguise Detection\Test Model\MMI-Det\models\yolo_test.py", line 160, in forward return self.forward_once(x, x2, profile) File "D:\OneDrive - Sejong University\NgheNhan_Master\Research\Facial Disguise Detection\Test Model\MMI-Det\models\yolo_test.py", line 187, in forward_once x = self.Enhance(x) # CEM File "C:\Users\Nghe_Nhan\anaconda3\envs\MMI_Det\lib\site-packages\torch\nn\modules\module.py", line 1931, in getattr raise AttributeError( AttributeError: 'Model' object has no attribute 'Enhance'