목록DNN/Detection (1)
The Beautiful Future
YoloV3
https://github.com/eriklindernoren/PyTorch-YOLOv3 코드분석 Grid Anchor 쿠다 사용사용 가능인지 보고 Float 타입선택 그리드 싸이즈는 출력의 높이 넓이, 입력 이미지 크기를 나눠서 스트라이드 크기를 지정 그리드 x, y를 arange를 사용해서 0~g까지 만들고 반복해서 그리드로 만듬, 그리드 shape은 11gg anchor의 넓이 높이 크기를 스트라이드 대비한 크기로 만들어 줌, 즉 원래 anchor의 크기는 입력 이미지에 대비한 크기였다. 리스트 튜플로 w, h 순으로 담기고 1n11 shape으로 w와 h를 나눔. def compute_grid_offsets(self, grid_size, cuda=True): self.grid_size = grid..
DNN/Detection
2020. 1. 13. 12:03