https://github.com/strcoder4007/yolo-v1-tf1

YOLO V1 implementation in Tensorflow 1 using Darknet-19 as backbone
https://github.com/strcoder4007/yolo-v1-tf1

cs230 darknet deep-learning tensorflow yolo

Last synced: 2 months ago
JSON representation

YOLO V1 implementation in Tensorflow 1 using Darknet-19 as backbone

• Host: GitHub
• URL: https://github.com/strcoder4007/yolo-v1-tf1
• Owner: strcoder4007
• License: mit
• Created: 2024-03-15T13:20:39.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2024-03-19T04:21:03.000Z (about 1 year ago)
• Last Synced: 2025-01-11T08:30:13.180Z (4 months ago)
• Topics: cs230, darknet, deep-learning, tensorflow, yolo
• Language: Python
• Homepage:
• Size: 2.37 MB
• Stars: 0
• Watchers: 3
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# YOLO-V1-TF1

This repository contains an implementation of YOLO v1 (You Only Look Once version 1) using TensorFlow 1. YOLO is a popular object detection algorithm known for its speed and accuracy using Darknet-19 as backbone.

![](/images/prob_map.png) ![](/images/anchor_map.png)

## Features

- Implementation of the YOLO v1 algorithm

- Training and inference scripts

- Pre-trained weights for quick start

- Evaluation metrics for model performance

## Installation

- Clone YOLO-V1-TF1 repository

```

git clone https://github.com/strcoder4007/YOLO-V1-TF1.git

cd YOLO-V1-TF1

```

- Run the download script to Download Pascal VOC dataset, and create correct directories

```

./download_data.sh

```

- Download YOLO_small weight file and put it in data/weight

## Configuring Hyperparameters 

Modify configuration in yolo/config.py

## Training

```

python train.py

```

## Test

```

python test.py

```