Deep CV

x = torch.rand(64, 32, 3) y = x.permute(2, 0, 1) y.size() --->[3, 64, 32]

CUDA 버전에 따라서 PyTorch 버전을 맞춰줘야 CUDA를 사용할 수 있었다. $ nvcc -V Cuda tool의 버전을 확인하고 버전에 맞춰 아래 사이트의 코드로 Pytorch를 install 해줬더니 해결됐다. https://pytorch.org/ PyTorch An open source machine learning framework that accelerates the path from research prototyping to production deployment. pytorch.org 옛날 pytorch 버전 설치는 아래 링크 https://pytorch.org/get-started/previous-versions/ ++ 추가) 3090, CUDA 11.3 version 사용시 pyt..

torch.clamp(input, min=None, max=None, *, out=None) min과 Max를 정하고, min보다 작으면 min으로 Max 보다 크면 Max로 제한 합니다. Parameters input (Tensor) – the input tensor. min (Number or Tensor, optional) – lower-bound of the range to be clamped to max (Number or Tensor, optional) – upper-bound of the range to be clamped to Example >>> a = torch.randn(4) >>> a tensor([-1.7120, 0.1734, -0.0478, -0.0922]) >>> torch.c..

torch.linspace(start, end, steps, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) start와 end를 사이가 균일하게 나눠진 steps 크기의 tensor를 생성합니다. Parameters start (float) – the starting value for the set of points end (float) – the ending value for the set of points steps (int) – size of the constructed tensor Example >>> torch.linspace(3, 10, steps=5) tensor([ 3.0000, 4.7500,..

작품 개발 의도 이런 건 생략하고 간단하게 어떻게 만들었는지 가겠습니다. 레츠고~  모습은 이렇습니다. 처음 구상할 때 쓰레기통을 생각해서 라벨이 부착돼 있으면 뱉는 걸 생각해서 생김새가 이렇습니다.상품성을 위해 자동 선별기로 구현 방향을 바꿔 컨베이어 벨트를 추가했습니다. 모터, 초음파 제어는 아두이노를 사용했습니다.  작동 알고리즘1. 컨베이어 벨트가 작동되고, 초음파에서 거리를 측정해서 일정 거리 이하가 되면 잠깐 더 작동 후에 멈춥니다.2. PC에게 신호를 보내면 PC에서 웹캠 모듈을 통해 사진을 촬영하고, 전처리 후 prediction을 합니다.3. 결과에 따라 이름을 붙여 사진을 저장합니다.4. 결과를 신호를 다시 아두이노에 보냅니다. 5. 신호에 따라 서보모터 방향을 다르게 돌립니다. 회로도..