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

Face uncertainty quantification or estimation using PyTorch.

Scalable implementation of Lee / Mykland (2012) and Ait-Sahalia / Jacod (2012) Jump tests for noisy high frequency data

[ICCV 2021 Oral] Mining Latent Classes for Few-shot Segmentation

Learning Multiresolution Matrix Factorization and its Wavelet Networks on Graphs

3.8% and 18.3% on CIFAR-10 and CIFAR-100

A python library to artfully visualize Factorio Blueprints and an interactive web demo for using it.

How will electric vehicles affect traffic congestion and energy consumption: an integrated modelling approach

Predict multi paths to a moving person depending on his trajectory history.

TLXZoo - Pre-trained models based on TensorLayerX

Directed Greybox Fuzzing with AFL

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

Sample Prior Guided Robust Model Learning to Suppress Noisy Labels

Code for the Paper "Diffusion Models for Handwriting Generation"

A machine learning malware analysis framework for Android apps.

Syllabic Quantity Patterns as Rhythmic Features for Latin Authorship Attribution

A python library for face detection and features extraction based on mediapipe library

Keyword2Text This repository contains the code of the paper: "A Plug-and-Play Method for Controlled Text Generation"

PyTorch implementation of paper "Neural Scene Flow Fields for Space-Time View Synthesis of Dynamic Scenes", CVPR 2021

The code for Expectation-Maximization Attention Networks for Semantic Segmentation (ICCV'2019 Oral)

Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering