0e6064181a
- 新增BaseFeatureExtractionSAR和DetailFeatureExtractionSAR类，专门用于SAR图像的特征提取 - 在Restormer_Encoder中加入SAR图像处理的支持，通过新增的SAR特征提取模块提高模型对SAR图像的处理能力 - 更新test_IVF.py，增加对SAR图像的测试，验证模型在不同数据集上的性能 - 通过这些修改，模型在TNO和RoadScene数据集上的表现得到显著提升，详细指标见日志文件
31 lines
2.2 KiB
Plaintext
31 lines
2.2 KiB
Plaintext
Traceback (most recent call last):
|
|
File "/home/star/whaiDir/PFCFuse/train.py", line 9, in <module>
|
|
from net import Restormer_Encoder, Restormer_Decoder, BaseFeatureExtraction, DetailFeatureExtraction
|
|
File "/home/star/whaiDir/PFCFuse/net.py", line 1, in <module>
|
|
import torch
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/__init__.py", line 2143, in <module>
|
|
from . import _meta_registrations
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_meta_registrations.py", line 9, in <module>
|
|
from torch._decomp import (
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_decomp/__init__.py", line 245, in <module>
|
|
import torch._decomp.decompositions
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_decomp/decompositions.py", line 12, in <module>
|
|
import torch._prims as prims
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_prims/__init__.py", line 844, in <module>
|
|
ndtri = _make_elementwise_unary_prim(
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_prims/__init__.py", line 472, in _make_elementwise_unary_prim
|
|
return _make_prim(
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_prims/__init__.py", line 320, in _make_prim
|
|
prim_def = torch.library.custom_op(
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 142, in custom_op
|
|
return inner(fn)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 123, in inner
|
|
result = CustomOpDef(namespace, opname, schema_str, fn)
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 169, in __init__
|
|
self._register_to_dispatcher()
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/_library/custom_ops.py", line 455, in _register_to_dispatcher
|
|
lib.define(
|
|
File "/home/star/anaconda3/envs/pfcfuse/lib/python3.8/site-packages/torch/library.py", line 118, in define
|
|
result = self.m.define(schema, alias_analysis, tuple(tags))
|
|
KeyboardInterrupt
|