Skip to content

ngvananh2508/Object_detection_via_PyTorch_based_YOLOv1_paper_model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 

Repository files navigation

YOLOv1-based model for object detection using PyTorch framework

This repository contains a script and datasets to train a model for object detection tasks. This model was constructed based on YOLOv1 paper with pretrained EfficientNetB1 model using PyTorch framework because TensorFlow temporarily has an internal issue which impedes us to train this. This is the link lead to the saved location of the weights of the trained model: https://drive.google.com/file/d/1O9U2vyqqJPhDfQnlnprSmxRpNu69n3RP/view?usp=sharing

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published