(pytorch) torch에서 사용할 수 있는 3d classification models 정리

Torch에서 제공하는 models은 2가지 경로로 가져올 수 있음

1. torchvision

• Document : https://pytorch.org/vision/main/models.html#video-classification

Models and pre-trained weights — Torchvision main documentation

• torchvision libarary를 설치 해야함 (보통 pytorch 설치 시, 같이 설치하는 편이긴함)

2. torch.hub

Document : https://pytorch.org/hub/research-models

PyTorch

• torch.hub로 접근해서 모델을 직접 다운받아오는 형식
• 위 사이트에서는 아래의 3가지 모델을 3d classification으로 사용할 수 있음
  • Resnet 3D - https://pytorch.org/hub/facebookresearch_pytorchvideo_resnet/
  • Slowfast - https://pytorch.org/hub/facebookresearch_pytorchvideo_slowfast/
  • X3D - https://pytorch.org/hub/facebookresearch_pytorchvideo_x3d/

 

모델의 자세한 사용방법은 document를 참고하며, 추후 시간과 여력과 체력이 된다면 여기에 튜토리얼을 간단하게 작성해보고자 합니다...