MMSegmentation-Installationskonfiguration
Eine grundlegende Umgebungskonfiguration
Installieren Sie zuerst Pytorch
pip3 install install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
Verwenden Sie dann mim, um MMCV zu installieren.
Die oben genannten Schritte sind nicht beliebt, und führen Sie dann die folgenden Vorgänge aus: Installieren Sie andere erforderliche Bibliotheken
pip3 install opencv-python pillow matplotlib seaborn tqdm pytorch-lightning 'mmdet>=3.0.0rc1' -i https://pypi.tuna.tsinghua.edu.cn/simple
2. Installieren Sie MMSegmentation
Überprüfen Sie zunächst, ob lokal eine MMSegmentierung vorhanden ist, und löschen Sie diese gegebenenfalls.
# 删掉原有的 mmsegmentation 文件夹(如有)
rm -rf mmsegmentation
Klonen Sie es dann von der offiziellen Website.
Nachdem der Download abgeschlossen ist, geben wir den Ordner ein, den wir gerade heruntergeladen haben, und kompilieren es: (Ich habe es viele Male geklont und werde immer vergewaltigt)
Nach dem Download können Sie überprüfen, ob die Installation erfolgreich war, indem Sie den folgenden Befehl ausführen:
3. Verwenden von MMSegmentierung
Wir können die vorab trainierte Modell- und Gewichtsdatei Model Zoo hier verwenden
: https://github.com/open-mmlab/mmsegmentation/blob/master/docs/en/model_zoo.md
Hier können Sie die Konfigurationsdatei herunterladen und vorab Das Trainingsmodell lässt sich einfach anwenden.
Für Experimente nutzen wir PSPNet