# yolov1_pytorch

**Repository Path**: tafkap_1981/yolov1_pytorch

## Basic Information

- **Project Name**: yolov1_pytorch
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-06-22
- **Last Updated**: 2024-06-22

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# yolov1_pytorch
Train and test YOLOv1 in PyTorch  
  
This code is a combination of two excellent YOLO-v1 in PyTorch repositories:  
1. https://github.com/motokimura/yolo_v1_pytorch  
2. https://github.com/abeardear/pytorch-YOLO-v1  

Specifically, the model training and testing procedure comes from (1),  
and model backbone from (2)- I modified the last few layers to enable a final feature map of (7,7) instead of (14,14).  
The model backbone is ResNet50.  
  
There are some modifications in terms of recording train and test history.  
Use train.py to train, and detect.py to perform detection with desired images.  

![](contents/result.jpg)