Implementação do Módulo Backbone do YOLOv5 para Classificação de Imagens
1. Configuração do Disopsitivo
import torch
import torch.nn as nn
import torchvision.transforms as transforms
from torchvision import datasets
import pathlib, warnings
warnings.filterwarnings("ignore")
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
2. Carregamento dos Dados
data_dir = path ...
Publicado em 8-22 02:57