PyPI 카운터 | |
PyPI 카운터 | |
PyPI 카운터 | |
Github 별 |
공지영상
데모 https://deepnote.com/project/QuickAI-1r_4zvlyQMa2USJrIvB-kA/%2Fnotebook.ipynb
더 발전된 기계 학습을 시작하면서 EfficientNet과 같은 유명한 신경망 아키텍처가 어떻게 놀라운 일을 하는지 알아보기 시작했습니다. 그러나 해결하고 싶은 문제에 대해 이러한 아키텍처를 구현하려고 시도했을 때 이러한 아키텍처를 구현하고 빠르게 실험하는 것이 매우 쉽지 않다는 것을 깨달았습니다. 이것이 바로 QuickAI가 등장한 이유입니다. 이를 통해 많은 모델 아키텍처를 빠르고 쉽게 실험할 수 있습니다.
Tensorflow, PyTorch, Sklearn, Matplotlib, Numpy 및 Hugging Face Transformers. 해당 웹사이트의 지침에 따라 TensorFlow 및 PyTorch를 설치해야 합니다.
위의 모든 종속성을 설정하지 않으려면 QuickAI Docker 컨테이너를 사용할 수 있습니다.
먼저 컨테이너를 가져옵니다. docker pull geekjr/quickai
그런 다음 실행하십시오.
CPU(Apple Silicon Mac에서는 --platform linux/amd64
플래그와 Rosetta 2 설치가 필요함): docker run -it geekjr/quickai bash
GPU: docker run --gpus all -it geekjr/quickai bash
QuickAI는 수십 줄의 코드를 1~2줄로 줄일 수 있습니다. 이는 빠른 실험을 매우 쉽고 깔끔하게 만듭니다. 예를 들어, 자신의 데이터 세트에서 EfficientNet을 교육하려면 데이터 로드, 전처리, 모델 정의 및 교육 코드(많은 줄의 코드)를 수동으로 작성해야 합니다. 반면 QuickAI를 사용하면 이러한 모든 단계가 단 1~2줄의 코드만으로 자동으로 수행됩니다.
pip install quickAI
자세한 내용은 예제 폴더를 참조하세요. YOLOV4의 경우 여기에서 가중치를 다운로드할 수 있습니다. 전체 문서는 저장소의 위키 섹션에 있습니다.
버그가 발견되면 수정될 수 있도록 새 문제를 열어주세요. 일반적인 질문이 있는 경우 토론 섹션을 이용하세요.
YOLO 구현을 위한 대부분의 코드는 "The AI Guy's" tensorflow-yolov4-tflite 및 YOLOv4-Cloud-Tutorial 저장소에서 가져왔습니다. 이것이 없으면 YOLO 구현이 불가능합니다. 감사합니다!