목록vision (6)

끊임없이 부단히

[mmrotate] 나만의 데이터 셋으로 rotated RetinaNet 학습하기

이번 포스팅에서는 mmrotate에서 제공하는 데이터 셋이 아닌 임의의 데이터 셋을 이용하여 모델을 학습하려 합니다. 새로운 데이터 셋으로는 앞선 포스팅에서 공부했던 COCO 데이터를 이용합니다. 1. DOTA 데이터 형태로 변환 먼저, COCO 데이터 셋을 DOTA 데이터 형태로 변환합니다. COCO는 bbox가 x, y, w, h 형태이므로 이를 x1, y1, x2, y2, x3, y3, x4, y4 형태로 바꾸는 함수를 작성합니다. 여기서 COCO의 x, y는 왼쪽 위(left-top)의 꼭지점 위치이며 이미지의 좌상단을 원점으로 합니다. DOTA도 x1, y1를 왼쪽 위로 하며 나머지들은 시계 방향으로 꼭지점들을 의미합니다. 추가로 category정보와 difficult정보(1: 어려움, 0: 쉬..
인공지능 2022. 10. 14. 00:30
[mmrotate] mmrotate로 rotated RetinaNet 학습하기

지난 포스팅에 이어서 mmrotate를 이용하여 rotated RetinaNet을 학습하려 합니다. 마찬가지로 아래의 DOTA 홈페이지에서 학습용 데이터를 다운받아 줍니다. 참고로, 버전 1, 1.5, 2가 있는데 버전 1을 다운받아야 합니다. (1.5이상을 쓸 경우 'container crane'이 없다는 오류가 납니다) 다운받은 데이터를 각각 data/DOTA/train & val 폴더 내의 images & labelTxt 폴더에 정리해줍니다. https://captain-whu.github.io/DOTA/dataset.html DOTA DOTA 2 --> DOTA A Large-Scale Benchmark and Challenges for Object Detection in Aerial Images..
인공지능 2022. 10. 13. 09:58
[mmrotate] mmrotate로 rotated RetinaNet 돌려보기

이번 포스팅에서는 open-mmlab의 mmrotate를 실행하려 합니다. 먼저 지난 포스팅에서 만든 환경에서 이어서 진행힙니다. https://fictitious.tistory.com/4 [mmrotate] open-mmlab의 mmrotate docker로 실행하기 이번 포스팅에서는 open-mmlab의 mmrotate를 실행하려 합니다. 우선 git clone을 해주고 폴더 이동을 합니다. git clone https://github.com/open-mmlab/mmrotate.git cd mmrotate 참고로 github 주소는 아래와 같.. fictitious.tistory.com mmrotate는 많은 object detection (특히, 회전된 bounding box 기반의) 기법들을 제공..
인공지능 2022. 10. 12. 00:50
[mmrotate] open-mmlab의 mmrotate docker로 실행하기

이번 포스팅에서는 open-mmlab의 mmrotate를 실행하려 합니다. 우선 git clone을 해주고 폴더 이동을 합니다. git clone https://github.com/open-mmlab/mmrotate.git cd mmrotate 참고로 github 주소는 아래와 같습니다. https://github.com/open-mmlab/mmrotate GitHub - open-mmlab/mmrotate: OpenMMLab Rotated Object Detection Toolbox and Benchmark OpenMMLab Rotated Object Detection Toolbox and Benchmark - GitHub - open-mmlab/mmrotate: OpenMMLab Rotated Obj..
인공지능 2022. 7. 26. 04:05
[Dataset] PyTorch의 Dataset 활용하기

PyTorch와 Dataset 이번 포스팅에서는 학습을 위해 간단한 Dataset을 만들려고 합니다. PyTorch의 torch.utils.data.DataLoader와 torch.utils.data.Dataset는 학습을 위한 데이터를 더욱 편리하게 관리할 수 있게 합니다. COCO 데이터 셋을 이용하여 이전 포스팅과 내용이 이어지니 참고하시면 도움이 될 것 같습니다. https://fictitious.tistory.com/2 [Python] Pycocotools를 이용한 COCO Dataset 확인하기 COCO Dataset 이번 포스팅은 Pycocotools를 이용하여 coco dataset에 데이터들이 어떤 형태로 있는지 확인합니다. 데이터 셋은 아래 링크에서 다운 받을 수 있으며 저는 /host..
인공지능 2022. 7. 22. 02:28
[Dataset] Pycocotools를 이용한 COCO Dataset 확인하기

COCO Dataset 이번 포스팅은 Pycocotools를 이용하여 coco dataset에 데이터들이 어떤 형태로 있는지 확인합니다. 데이터 셋은 아래 링크에서 다운 받을 수 있으며 저는 /hostDir/dataSet/images 와 /hostDir/dataSet/annotation 폴더에 다운받아 코드를 실행합니다. https://cocodataset.org/#download COCO - Common Objects in Context cocodataset.org 1. COCO 초기화 COCO 데이터 셋은 사진들과 json 파일로 구성되어 있습니다. 우선 필요로하는 모듈들을 import하고 json 파일 경로를 COCO 객체의 입력으로 넣어 coco를 생성합니다. import os import tor..
인공지능 2022. 7. 20. 04:06
Prev 1 Next