Iranian Cars Detection using Yolov5

Train

git clone https://github.com/ultralytics/yolov5
cd yolov5
pip install -r requirements.txt

2- Dataset

../Dataset/images/im0.jpg  # image
../Dataset/labels/im0.txt  # label

3- Create dataset.yaml

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]

path: ../Dataset  # dataset root dir
train: ../Dataset/Train/images  # train images (relative to 'path') 128 images
val: ../Dataset/Val/images  # val images (relative to 'path') 128 images

# Classes
nc: 5  # number of classes
names: ['iranKhodro_dena', 'kia_cerato', 'mazda_3', 'peugeot_206', 'saipa_saina']  # class names

You can change this sections in Yolov5/data/coco128.yaml.

4- Train YOLOv5s on Dataset for 30 epochs using following command:

$ python train.py --img 640 --batch 16 --epochs 30 --data coco128.yaml --weights yolov5s.pt

Inference

!python /yolov5/detect.py --weights /runs/train/exp13/weights/last.pt --img 244 --conf 0.4 --source /data/saipa_saina88.jpg

Iranian Cars Detection using Yolov5s, PyTorch

Related tags

Overview

Iranian Cars Detection using Yolov5

Train

Inference

Owner

Nahid Ebrahimian

Convolutional neural network web app trained to track our infant’s sleep schedule using our Google Nest camera.

Python script for performing depth completion from sparse depth and rgb images using the msg_chn_wacv20. model in Tensorflow Lite.

Bayesian Image Reconstruction using Deep Generative Models

Official Implementation of HRDA: Context-Aware High-Resolution Domain-Adaptive Semantic Segmentation

(ICCV 2021) PyTorch implementation of Paper "Progressive Correspondence Pruning by Consensus Learning"

Neural Dynamic Policies for End-to-End Sensorimotor Learning

RATCHET is a Medical Transformer for Chest X-ray Diagnosis and Reporting

Pytorch reimplement of the paper "A Novel Cascade Binary Tagging Framework for Relational Triple Extraction" ACL2020. The original code is written in keras.

(EI 2022) Controllable Confidence-Based Image Denoising

Music Classification: Beyond Supervised Learning, Towards Real-world Applications

Unofficial implementation of One-Shot Free-View Neural Talking Head Synthesis

This is the code used in the paper "Entity Embeddings of Categorical Variables".

Code for "NeuralRecon: Real-Time Coherent 3D Reconstruction from Monocular Video", CVPR 2021 oral

This code reproduces the results of the paper, "Measuring Data Leakage in Machine-Learning Models with Fisher Information"

A PyTorch implementation of "Multi-Scale Contrastive Siamese Networks for Self-Supervised Graph Representation Learning", IJCAI-21

The code for the NeurIPS 2021 paper "A Unified View of cGANs with and without Classifiers".

Only works with the dashboard version / branch of jesse

Setup and customize deep learning environment in seconds.

MILK: Machine Learning Toolkit

KGDet: Keypoint-Guided Fashion Detection (AAAI 2021)