-
Notifications
You must be signed in to change notification settings - Fork 159
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
New test.py #53
Comments
hello, i use the code ,but error: `python new_test.py --save net_model/_8514_CUB/weights_20_0_1.0000_1.0000.pth --acc_report True |
check out your num_classes |
thanks,i have solve it |
@Lamborghini1709 @yunchangxiaoguan 你好,能不能分享下训练好的cub模型,这个模型是真的需要硬件。小batch训练太慢。 |
@yunchangxiaoguan 您好请问这个size mismatch for classifier_swap.weight:问题您如何解决的呢,谢谢 |
检查你的输出类别数 num_classes |
#coding=utf-8
import os
import json
import csv
import argparse
import pandas as pd
import numpy as np
from math import ceil
from tqdm import tqdm
import pickle
import shutil
import torch
import torch.nn as nn
from torch.autograd import Variable
from torch.nn import CrossEntropyLoss
from torchvision import datasets, models
import torch.backends.cudnn as cudnn
import torch.nn.functional as F
from transforms import transforms
from models.LoadModel import MainModel
from utils.dataset_DCL import collate_fn4train, collate_fn4test, collate_fn4val, dataset
from config import LoadConfig, load_data_transformers
from utils.test_tool import set_text, save_multi_img, cls_base_acc
import pdb
os.environ['CUDA_DEVICE_ORDRE'] = 'PCI_BUS_ID'
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1,2,3'
def parse_args():
parser = argparse.ArgumentParser(description='dcl parameters')
parser.add_argument('--data', dest='dataset',
default='CUB', type=str)
parser.add_argument('--backbone', dest='backbone',
default='resnet50', type=str)
parser.add_argument('--b', dest='batch_size',
default=16, type=int)
parser.add_argument('--nw', dest='num_workers',
default=16, type=int)
parser.add_argument('--ver', dest='version',
default='test', type=str)
parser.add_argument('--save', dest='resume',
default=None, type=str)
parser.add_argument('--size', dest='resize_resolution',
default=512, type=int)
parser.add_argument('--crop', dest='crop_resolution',
default=448, type=int)
parser.add_argument('--ss', dest='save_suffix',
default=None, type=str)
parser.add_argument('--acc_report', dest='acc_report',
action='store_true')
parser.add_argument('--swap_num', default=[7, 7],
nargs=2, metavar=('swap1', 'swap2'),
type=int, help='specify a range')
args = parser.parse_args()
return args
if name == 'main':
args = parse_args()
print(args)
# if args.submit:
# args.version = 'test'
# if args.save_suffix == '':
# raise Exception('**** miss --ss save suffix is needed. ')
args.version = 'test'
Config = LoadConfig(args, args.version)
transformers = load_data_transformers(args.resize_resolution, args.crop_resolution, args.swap_num)
data_set = dataset(Config,
anno=Config.val_anno if args.version == 'val' else Config.test_anno ,
# unswap=transformers["None"],
swap=transformers["None"],
totensor=transformers['test_totensor'],
test=True)
run test : python test.py --save ./net_model/training_descibe_72721_CUB/weights_36_4999_0.8608_0.9998.pth --acc_report
The text was updated successfully, but these errors were encountered: