반응형 분류 전체보기40 Python setup.py 에서 닌자 관련 에러 날 때 가끔 setuptools에서 이상한 에러가 날 때가 있는데, 주로 컴파일러 문제일 때가 많다. cmake도 에러가 자주 나지만 닌자에서도 에러를 자주 보게 되어서, 닌자랑 cmake랑 둘 다 시도해보고 싶은 상황이 있게 된다. from setuptools import setup from torch.utils.cpp_extension import BuildExtension, CUDAExtension setup( name='roi_align_cuda', ext_modules=[ CUDAExtension('roi_align_cuda', [ 'src/roi_align_cuda.cpp', 'src/roi_align_kernel.cu', ]), ], cmdclass={'build_ext': BuildExtensio.. 2021. 4. 13. 파이썬에서 로그 한 줄에 계속 프린팅하기 파이썬에서 루프 진행사항을 프린트 할 때, 프린팅 단위가 너무 크면 어디까지 됐는지 감이 안 오고, 너무 짧으면 화면을 다 잡아먹어서 오히려 진행사항을 보기가 어려운 상황이 생긴다. tqdm 같은 패키지를 쓸 수 있지만, 굳이 프로그래스 바 수준까지는 필요 없다면 print("\r", sep=' ', end='', flush=True) 위 예시 및 코드블럭과 같이 프린트 함수에 옵션만 조정해도 괜찮은 결과를 얻을 수 있다. 2021. 4. 12. torch.nn.Parameter 에 관해서 Pytorch 에는 Parameter라는 모듈이 있는데, 얘는 레이어가 아니라 말 그대로 파라미터 값만을 가지고 있는 놈이다. class Actor(nn.Module): def __init__(self, num_inputs, num_outputs, continuous=True, shared=False): self.num_inputs = num_inputs self.num_outputs = num_outputs super(Actor, self).__init__() self.fc1 = nn.Linear(num_inputs, hp.hidden) self.fc2 = nn.Linear(hp.hidden, hp.hidden) self.fc3 = nn.Linear(hp.hidden, num_outputs) self... 2021. 4. 7. detectron2 에서 Faster R-CNN RPN에 GradCAM 붙이기 기본적인 코드 세팅은 github.com/yizt/Grad-CAM.pytorch 를 참조한다. 이 외에 detectron2 설치폴더 밑 modeling/proposal_generator/rpn.py 에서 RPN 클래스 하부 predict_proposals 함수에서 def predict_proposals( self, anchors: List[Boxes], pred_objectness_logits: List[torch.Tensor], pred_anchor_deltas: List[torch.Tensor], image_sizes: List[Tuple[int, int]], ): """ Decode all the predicted box regression deltas to proposals. Find the to.. 2021. 3. 30. 이전 1 ··· 6 7 8 9 10 다음
