ImportError: cannot import name '_NewEmptyTensorOp' from 'torchvision.ops.misc'

■ Sparse-rcnn 학습 및 추론 동작 테스트

해당 모델을 개발중인 MLOPS 서비스에 연동하려고 하는데...

Sparse-rcnn github에 있는 Dockerfile을 활용하면

■ sudo apt-get update 시 NO_PUBKEY

에러가 발생한다. 해당 문제는 다음과 같이 해결할 수 있다고 하는데, 결과적으로 나는 끝까지 안됐다.
참고 : https://eehoeskrap.tistory.com/454

■ detectron2 동작 테스트

그래서 그냥 detectron2의 모델을 ngc pytorch docker에 있는 22.04 버전에 설치해주었다.
그리고 아래 페이지를 참고해서 demo와 train_net.py가 정상적으로 동작하는 것을 체크했다.
참고 : https://detectron2-kr.netlify.app/tutorials/getting_started.html#inference

※ 주의할 점:

1. model weight파일 따로 다운로드 받기 https://github.com/facebookresearch/detectron2/blob/main/MODEL_ZOO.md

2. --opt MODEL.WEIGHTS [다운받은 모델]

3. 학습 시, coco 경로 export 설정

■ Sparse-rcnn torchvision 버전관련 에러

if float(torchvision.__version__[:3]) < 0.7:
    from torchvision.ops import _new_empty_tensor
    from torchvision.ops.misc import _output_size

정말 어이가 없지만,,, 해당 코드 주석처리하면 잘 돌아간다. torchvision 0.8, 0.9버전 돌렸으면 문제없이 됐을텐데,,, 역시 운빨게임...