deepinv/deepinv

DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning

해결하는 문제

DeepInverse는 이미징 역문제를 해결하는 과제를 다룹니다. 다양한 이미징 도메인에서 딥러닝 연구를 가속화하기 위한 모듈식 프레임워크를 제공하여 재현성을 높이고 새로운 실무자의 진입 장벽을 낮춥니다.

작동 방식

이 라이브러리는 PyTorch를 기반으로 구축되었으며 다음과 같은 주요 구성 요소를 제공합니다:

  • 사전 정의된 이미징 연산자의 대규모 프레임워크.
  • 사전 학습된 재구성 모델 및 노이즈 제거기를 포함한 최첨단 딥 뉴럴 네트워크.
  • 플러그 앤 플레이 방식의 복원, 최적화 및 unfolded 아키텍처를 위한 프레임워크.
  • 역문제를 위한 특화된 학습 손실 함수.
  • 불확실성 정량화를 위한 샘플링 알고리즘 및 확산 모델.
  • 역문제를 위한 데이터셋 구축 전용 프레임워크.

대상

역문제를 위한 딥러닝 알고리즘을 구현하고 테스트하기 위한 표준화된 모듈식 방법을 필요로 하는 이미징 도메인의 연구자 및 실무자를 위해 설계되었습니다.

주요 특징

  • 공식 PyTorch 생태계의 일부
  • 플러그 앤 플레이 방식의 복원 및 unfolded 아키텍처 지원
  • 불확실성 정량화를 위한 확산 모델 포함
  • 특화된 데이터셋 구축을 위한 프레임워크 제공