jiantenggei / torch-classification
用来自己训练和复现分类模型
☆18Updated 2 years ago
Alternatives and similar repositories for torch-classification:
Users that are interested in torch-classification are comparing it to the libraries listed below
- pytorch resnet image-classification☆19Updated 2 years ago
- 利用pytorch实现图像分类的一个完整的代码,训练,预测,TTA,模型融合,模型部署,cnn提取特征,svm或者随机森林等进行分类,模型蒸馏,一个完整的代码☆28Updated 4 years ago
- 这是一个cyclegan-pytorch的源码,可以用于训练自己的模型。☆42Updated 2 years ago
- 基于PyTorch框架实现的图像分类网络☆79Updated 4 years ago
- 基于Swin-transformer训练图像分类并部署web端☆86Updated 2 years ago
- 玻璃绝缘子缺陷检测☆32Updated 4 years ago
- Yet another coco2voc.py, convert MSCOCO json to PASCAL VOC xml in Object Detection.☆25Updated 4 years ago
- demo project of <A Surface Defect Detection Method Based on Positive Samples>, deployed in pytorch☆39Updated 4 years ago
- I tried to apply the CAM algorithm to YOLOv4 and it worked.☆61Updated 3 years ago
- 基于YOLOv5和PSPNet的实时目标检测和语义分割系统☆40Updated 2 years ago
- 深度学习 注意力机制模块 时间 空间注意力通道☆29Updated last year
- generate randomly defects on positive samples and restore them with GAN.☆40Updated 4 years ago
- Implementation of Gather-Excite Network based on Mindspore and pytorch☆16Updated 2 years ago
- 布匹缺陷识别练习赛☆45Updated 4 years ago
- ☆34Updated last year
- (1) Purpose: A weakly supervised surface defect detection model using image-level labels for simultaneous classification and segmentation…☆22Updated last year
- ☆84Updated 10 months ago
- This is the GC10-DET datasets of the upcoming paper " Deep Metallic Surface Defect Detection: the New Benchmark and Detection Network" Th…☆64Updated 3 years ago
- fpn_resnet、resnet-se、resnet-cbam☆33Updated 5 years ago
- object detection dataset conversion scripts☆23Updated 2 years ago
- Compound Multi-branch Feature Fusion for Real Image Restoration☆95Updated 7 months ago
- Implementation of augmentation for small object detection(填鸭) https://arxiv.org/pdf/1902.07296.pdf☆19Updated 4 years ago
- Cosine Annealing Warm Booting LR☆27Updated 3 years ago
- pytorch实现MOAT,可以在ImageNet或自己的数据集上训练,支持apex混合精度,各种图像增强技术☆29Updated 2 years ago
- 基于Pytorch实现优秀的自然图像分割框架!(包括FCN、U-Net和Deeplab)☆65Updated 3 years ago
- Implement CBAM:Convolutional Block Attention Module with TensorFlow☆27Updated 6 years ago
- This repository contains the implementation and the building blocks of GlideNet and Informed Convolution. This work is published at CVPR …☆28Updated 3 years ago
- CvPytorch is an open source COMPUTER VISION toolbox based on PyTorch.☆251Updated last year
- Official repository of paper titled "CAS-ViT: Convolutional Additive Self-attention Vision Transformers for Efficient Mobile Applications…☆55Updated 3 months ago
- ☆10Updated 2 years ago