Skip to content

Commit

Permalink
add badgev2
Browse files Browse the repository at this point in the history
  • Loading branch information
igorastashov committed Dec 11, 2023
1 parent 0e4efbf commit dabb874
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
## MLops project

[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
[![Build Status](https://github.com/igorastashov/MLops-project/actions/workflows/checks.yml/badge.svg)](https://github.com/igorastashov/MLops-project/actions/workflows/checks.yml)

**Pokemon image classification using Mobilenet_v2 model**

Astashov I.V., 2023.

[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
![Build Status](https://github.com/igorastashov/MLops-project/actions/workflows/Code-checks/badge.svg?branch=main)

This repository contains model, evaluation code and training code on dataset
from [kaggle](https://www.kaggle.com/datasets/lantian773030/pokemonclassification).
**If you would like to run pretrained model on your image see [(2) Quick start](https://github.com/igorastashov/MLops-project#2-quick-start)**.
Expand Down Expand Up @@ -66,5 +66,3 @@ python train.py

This repository borrows partially from [Isadrtdinov](https://github.com/isadrtdinov/intro-to-dl-hse/blob/2022-2023/seminars/201/seminar_04.ipynb), and [FUlyankin](https://github.com/FUlyankin/deep_learning_pytorch/tree/main/week08_fine_tuning) repositories.
Repository design taken from [v-goncharenko](https://github.com/v-goncharenko/data-science-template) and [PeterWang512](https://github.com/PeterWang512/CNNDetection).

![Build Status](https://github.com/igorastashov/MLops-project/actions/workflows/checks.yml/bage.svg?branch=main)

0 comments on commit dabb874

Please sign in to comment.