luke12121212/ship-classification-resnet18

Ship Classification AI — aplikacja lokalna

ResNet18 wytrenowany w Google Colab, 5 klas: Cargo, Military, Carrier, Cruise, Tankers.

Uruchomienie lokalne

1. Zainstaluj Python 3.10–3.12 (zalecana wersja zgodna z aktualnym PyTorch).

2. Rozpakuj katalog ship_classifier_local.

3. Otwórz terminal w katalogu i wpisz:

   python -m venv .venv
   

   Aktywuj środowisko:

   • Windows PowerShell: .venv\Scripts\Activate.ps1
   • macOS / Linux: source .venv/bin/activate

   Następnie:

   python -m pip install -r requirements.txt
   python app.py
   

4. Otwórz http://127.0.0.1:7860. Zakładka Predict klasyfikuje zdjęcia, a Metrics pokazuje wyniki.

Model i interfejs działają na Twoim komputerze, na CPU lub GPU. Wyniki testowe są w assets/. To klasyfikator całego obrazu, a nie detektor położenia statków.

Pochodzenie

Dataset: https://www.kaggle.com/datasets/arpitjain007/game-of-deep-learning-ship-datasets