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

Two-Stream Adaptive Graph Convolutional Networks for Skeleton-Based Action Recognition in CVPR19

SwinIR: Image Restoration Using Swin Transformer

disentanglement_lib is an open-source library for research on learning disentangled representations.

Few-NERD: Not Only a Few-shot NER Dataset

Keras documentation, hosted live at keras.io

This repository is for Competition for ML_data class

Official code implementation for "Personalized Federated Learning using Hypernetworks"

Adversarial Attacks on Probabilistic Autoregressive Forecasting Models.

Code release to accompany paper "Geometry-Aware Gradient Algorithms for Neural Architecture Search."

Code for the ICML 2021 paper: "ViLT: Vision-and-Language Transformer Without Convolution or Region Supervision"

Sequential Model-based Algorithm Configuration

LVI-SAM: Tightly-coupled Lidar-Visual-Inertial Odometry via Smoothing and Mapping

An implementation of the Contrast Predictive Coding (CPC) method to train audio features in an unsupervised fashion.

GazeScroller - Using Facial Movements to perform Hands-free Gesture on the system

Codecov coverage standard for Python

Traffic4D: Single View Reconstruction of Repetitious Activity Using Longitudinal Self-Supervision

Pytorch implementation of Supporting Clustering with Contrastive Learning, NAACL 2021

Self-Supervised Pillar Motion Learning for Autonomous Driving (CVPR 2021)

The implemetation of Dynamic Nerual Garments proposed in Siggraph Asia 2021

Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks