KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#dsh #dsh-plugin #ai-agents #developer-tools #codex #claude-code #mcp #rust #deepseek-harness #typescript #ai-agent #cli

gate-detection

★ 14 Open GitHub ↗

Underwater Object Detection using PyTorch

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg?style=flat-square)

# Underwater Object (Gate) Detection using Pytorch

## Overview

This repository created to train a model (Pre-trained Faster R-CNN ResNet-50 FPN Model)and detect an underwater object using pytorch API with the help of [Detecto](https://github.com/alankbi/detecto/).

See works with other frameworks: [tensorflow](https://github.com/enesdemirag/gate-detection/tree/tensorflow), [openCV](https://github.com/enesdemirag/gate-detection/tree/openCV).

Our team works on an Autonomous Underwater Vehicle to prepare for [Singapore AUV Challange](https://sauvc.org/) and the [RoboSub'20](https://robonation.org/programs/robosub/).

Check out our [website](http://auv.itu.edu.tr/) and [repo](https://gitlab.com/itu-auv).

> Tested on Ubuntu 18.04

## Getting Ready

#### 1. Install [Detecto](https://detecto.readthedocs.io/en/latest/usage/quickstart.html#installation)

```
pip3 install detecto
```

#### 2. Get your images for dataset.

#### 3. Use [labelimg](https://github.com/tzutalin/labelImg) to label images.

```
pip3 install labelImg
```

#### 4. Run below codes to convert xml files.

```
from detecto.utils import xml_to_csv
xml_to_csv('path_to_xml_folder/', 'labels.csv')
```

## Training

#### Local

- Train on your PC, run [this](training.py) script.

#### Server

- Train on a powerful machine, run [colab notebook](training.ipynb) on Google Colab.

## Testing

- Test using openCV, run [this](testing.py) script.

gate-detection

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README