# yolov1-pytorch

**Repository Path**: blake-john/yolov1-pytorch

## Basic Information

- **Project Name**: yolov1-pytorch
- **Description**: My implementation of yolov1 by using PyTorch
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-07-23
- **Last Updated**: 2024-07-23

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# YoloV1 PyTorch Implementation

This is a repository storing the implementation of yolov1.

## Structure

* `dataset.py` - with the class `YoloV1Dataset` to load the data
* `loss.py` - with the class `YoloV1Loss` to calculate the sum square loss
* `net.py` - with the class `YoloV1Net`, the neural network of yolov1
* `train.py` - perform the train stage
* `config.py` - the hyper parameters of the network