BiSeNet
BiSeNet copied to clipboard
BiSeNet based on pytorch
train.py里面并没有实现epoch_start_i函数判断的模块,所以当引用这个参数会报错。 因此目前来说,如果想接着上次迭代的epoch来进行训练的话,只能通过pre-train model 来操作
AttributeError: 'tuple' object has no attribute 'log_softmax'
`class DiceLoss(nn.Module): def __init__(self): super().__init__() self.epsilon = 1e-5` def forward(self, output, target): # print(output.shape) # print(target.shape) assert output.size() == target.size(), "'input' and 'target' must have the same shape" # 在classes上做softmax...
--------------------------------------------------------------------------- FileNotFoundError Traceback (most recent call last) in 5 #df 6 for site_name in df['SiteName'].unique(): ----> 7 df[df['SiteName'] == site_name].to_csv('{}.csv'.format(site_name)) ~\AppData\Local\Continuum\anaconda3\lib\site-packages\pandas\core\generic.py in to_csv(self, path_or_buf, sep, na_rep, float_format, columns, header, index,...
BiSeNet$ python demo.py Traceback (most recent call last): File "demo.py", line 81, in main(params) File "demo.py", line 60, in main model.module.load_state_dict(torch.load(args.checkpoint_path)) File "/home/keshi/anaconda3/lib/python3.7/site-packages/torch/serialization.py", line 381, in load f = open(f,...
Please see the function that come from utils.py: def one_hot_it_v11_dice(label, label_info): semantic_map = [] void = np.zeros(label.shape[:2]) for index, info in enumerate(label_info): color = label_info[info][:3] class_11 = label_info[info][3] if class_11...
Hello, I can't find this file.
Do you have any pretrained weights for COCO dataset. If not do you have any train.py or script to load the COCO dataset. Appreciate any help.