HiFuse

Este repositorio es la implementación oficial de HiFuse: Hierarchical Multi-Scale Feature Fusion Network for Medical Image Classification Autores: Xiangzuo Huo, Gang Sun, Shengwei Tian, Yan Wang, Long Yu, Jun Long, Wendong Zhang y Aolun Li. ¡Disfruta del código y encuentra su conveniencia para producir trabajos más impresionantes!

Visión General

Bloque HFF

Inspección Visual de HiFuse

Ejecución

• Requisitos:
• python3
• pytorch 1.10
• torchvision 0.11.1
• Entrenamiento:
• Prepara las imágenes requeridas y guárdalas en categorías, configura carpetas de imágenes para entrenamiento y validación respectivamente
• Ejecuta python train.py
• Reanudar entrenamiento:
• Modifica parser.add_argument('--RESUME', type=bool, default=True) en python train.py
• Ejecuta python train.py
• Pruebas:
• Ejecuta python test.py

TensorBoard

Ejecuta tensorboard --logdir runs --port 6006 para ver el progreso del entrenamiento

Referencias

Algunos códigos de este repositorio están tomados de:

• Swin Transformer
• ConvNeXt
• WZMIAOMIAO

Citación

Si encuentra que nuestro artículo/código es útil, por favor considere citar:

@article{huo2024hifuse,
  title={HiFuse: Hierarchical multi-scale feature fusion network for medical image classification},
  author={Huo, Xiangzuo and Sun, Gang and Tian, Shengwei and Wang, Yan and Yu, Long and Long, Jun and Zhang, Wendong and Li, Aolun},
  journal={Biomedical Signal Processing and Control},
  volume={87},
  pages={105534},
  year={2024},
  publisher={Elsevier}
}
@article{huo2025mm,
  title={MM-HiFuse: multi-modal multi-task hierarchical feature fusion for esophagus cancer staging and differentiation classification},
  author={Huo, Xiangzuo and Tian, Shengwei and Yu, Long and Zhang, Wendong and Li, Aolun and Yang, Qimeng and Song, Jinmiao},
  journal={Complex \& Intelligent Systems},
  volume={11},
  number={1},
  pages={1--12},
  year={2025},
  publisher={Springer}
}

Star History

--- Tranlated By Open Ai Tx | Last indexed: 2026-05-19 ---