![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
lykhahaha
ROS 机器人 计算机视觉
展开
-
读取zs数据 dataset 训练一维卷积模型
trainer.py from bdb import set_trace import torch from torch import nn from nets import ConvNet, LstmNet from dataset import tempDataset import os from torch.utils.data import Dataset, DataLoader class Trainer: def __init__(self, save_path, dataset.原创 2022-05-10 00:07:22 · 491 阅读 · 0 评论 -
test pytorch 测试模型绘图到图片
#coding=utf-8 import torch import numpy as np from torch.utils.data import DataLoader from torch.autograd import Variable import matplotlib.pyplot as plt import pandas as pd from model import myconvNet import os import cv2 import time from dataloader impo.原创 2022-01-17 21:38:47 · 1257 阅读 · 0 评论