python 93% acc. CNN Dogs Vs Cats ( Pytorch )

Last update: May 22, 2022

Overview

English | 简体中文（测试中...敬请期待）

Cnn-Classification-Dog-Vs-Cat 猫狗辨别

(pytorch版本) CNN Resnet18 的猫狗分类器，基于ResNet及其变体网路系列，对于一般的图像识别任务表现优异，模型精准度高达93%（小型样本）。

项目制作于本科大三学习 [認識人工智慧AI：企業人工智慧] 课堂期间，正好遇上本人对这方面感兴趣的阶段，所以选择了入门的深度学习项目练手，希望做为兴趣点激励自己学习！距离DDL只剩几天了，现阶段时间和精力有限，遂没有自建神经网络，只是利用了已训练的常用网络进行深度学习，以后找机会补上。

1 requirement

python3
matplotlib
numpy
pytorch
pandas
os
Images

2 Description of files

inputs: 包含猫狗训练和测试样本图片数据[下载地址]，经过特殊改良，其中训练集包含1000笔狗狗图片、1000笔猫咪图片，测试集包含100笔猫狗混合图片；
dog_cat_classcial.ipynb：主文件，训练后测试集精度约 93%
ckpt_resnet18_catdog.pth：基于CNN的预测模型
preds_resnet18.csv：预测后结果储存位置
true_test.csv：一笔正确的资料数据

3 Start training

印出部分训练集图片

在CNN（Resnet）的基础上进行深度学习
```
dog_cat_classcial.ipynb
```

4 Output prediction results

5 References

[1]. 猫狗图像数据来源： https://www.kaggle.com/c/dogs-vs-cats/data
[2]. 参考kaggle竞赛奖牌获得者项目 https://www.kaggle.com/uysimty/keras-cnn-dog-or-cat-classification

You might also like...

git《Beta R-CNN: Looking into Pedestrian Detection from Another Perspective》(NeurIPS 2020) GitHub:[fig3]

Beta R-CNN: Looking into Pedestrian Detection from Another Perspective This is the pytorch implementation of our paper "[Beta R-CNN: Looking into Pede

35 Sep 8, 2021

A spherical CNN for weather forecasting

DeepSphere-Weather - Deep Learning on the sphere for weather/climate applications. The code in this repository provides a scalable and flexible framew

47 Dec 25, 2022

BossNAS: Exploring Hybrid CNN-transformers with Block-wisely Self-supervised Neural Architecture Search

BossNAS This repository contains PyTorch evaluation code, retraining code and pretrained models of our paper: BossNAS: Exploring Hybrid CNN-transforme

127 Dec 26, 2022

《A-CNN: Annularly Convolutional Neural Networks on Point Clouds》(2019)

A-CNN: Annularly Convolutional Neural Networks on Point Clouds Created by Artem Komarichev, Zichun Zhong, Jing Hua from Department of Computer Science

44 Feb 24, 2022

Detection of drones using their thermal signatures from thermal camera through YOLO-V3 based CNN with modifications to encapsulate drone motion

Drone Detection using Thermal Signature This repository highlights the work for night-time drone detection using a using an Optris PI Lightweight ther

6 Dec 31, 2022

Segmentation and Identification of Vertebrae in CT Scans using CNN, k-means Clustering and k-NN

Segmentation and Identification of Vertebrae in CT Scans using CNN, k-means Clustering and k-NN If you use this code for your research, please cite ou

41 Dec 8, 2022

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

1.2k Jan 4, 2023

LiDAR R-CNN: An Efficient and Universal 3D Object Detector

LiDAR R-CNN: An Efficient and Universal 3D Object Detector Introduction This is the official code of LiDAR R-CNN: An Efficient and Universal 3D Object

295 Jan 5, 2023

Multistream CNN for Robust Acoustic Modeling

Multistream Convolutional Neural Network (CNN) A multistream CNN is a novel neural network architecture for robust acoustic modeling in speech recogni

37 Sep 21, 2022

python 93% acc. CNN Dogs Vs Cats ( Pytorch )

Related tags

Overview

Cnn-Classification-Dog-Vs-Cat 猫狗辨别

1 requirement

2 Description of files

3 Start training

印出部分训练集图片

在CNN（Resnet）的基础上进行深度学习

4 Output prediction results

5 References

You might also like...

git《Beta R-CNN: Looking into Pedestrian Detection from Another Perspective》(NeurIPS 2020) GitHub:[fig3]

A spherical CNN for weather forecasting

BossNAS: Exploring Hybrid CNN-transformers with Block-wisely Self-supervised Neural Architecture Search

《A-CNN: Annularly Convolutional Neural Networks on Point Clouds》(2019)

Detection of drones using their thermal signatures from thermal camera through YOLO-V3 based CNN with modifications to encapsulate drone motion

Segmentation and Identification of Vertebrae in CT Scans using CNN, k-means Clustering and k-NN

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

LiDAR R-CNN: An Efficient and Universal 3D Object Detector

Multistream CNN for Robust Acoustic Modeling

Owner

apple ye

Just Randoms Cats with python

Dogs classification with Deep Metric Learning using some popular losses

Code for paper ECCV 2020 paper: Who Left the Dogs Out? 3D Animal Reconstruction with Expectation Maximization in the Loop.

NFT-Price-Prediction-CNN - Using visual feature extraction, prices of NFTs are predicted via CNN (Alexnet and Resnet) architectures.

A faster pytorch implementation of faster r-cnn

Dense Deep Unfolding Network with 3D-CNN Prior for Snapshot Compressive Imaging, ICCV2021 [PyTorch Code]

A Pytorch Implementation of Source Data-free Domain Adaptation for a Faster R-CNN

An Ensemble of CNN (Python 3.5.1 Tensorflow 1.3 numpy 1.13)

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.

CoTr: Efficiently Bridging CNN and Transformer for 3D Medical Image Segmentation