科研平台运行BMN
源码:https://github.com/JJBOY/BMN-Boundary-Matching-Network
1.尝试运行main.py–mode train
python /workspace/BMN/main.py --mode train
问题1-导入module失败
No module named 'eval_proposal'
eval.py 相对路径改绝对路径
import sys
sys.path.append('./Evaluation')
为
import sys
sys.path.append('/workspace/BMN/Evaluation')
问题2-device id
main.py 87
model = torch.nn.DataParallel(model, device_ids=[0, 1]).cuda()
为
model = torch.nn.DataParallel(model, device_ids=[0]).cuda()
问题3-data缺文件?
opts.py文件名错误:
parser.add_argument('--video_info',type=str,default="/workspace/BMN/data/activitynet_annotations/video_info_new.csv")parser.add_argument('--video_anno',type=str,default="/workspace/BMN/data/activitynet_annotations/anet_anno_action.json")parser.add_argument('--temporal_scale',type=int,default=100)parser.add_argument('--feature_path',type=str,default="/workspace/BMN/data/activitynet_feature_cuhk/")
问题4 -特征数据还未导入
csv_mean_100下载好,解压放入data/activitynet_feature_cuhk
下载链接:https://github.com/wzmsltw/BSN-boundary-sensitive-network
问题5:pytorch版本为1.7
main.py
把scheduler.step()放在train()之后。
numworker修改为0
2.运行成功!
3.尝试运行main.py --mode inference
计算AR和recall
python /workspace/BMN/main.py --mode inference