- 博客(6)
- 收藏
- 关注
原创 (ICASSP 2024)RVDNet:面向真实世界的两阶段域自适应视频去雪网络
(1)提出了一种用于真实世界视频去雪的两阶段网络 RVDNet。 RVDNet 的第一阶段用于提取空间域特征。第二阶段利用时空信息去除视频中的雪花。(2)通过多阶段域对齐,将无监督域自适应(DA)的思想融入到视频 去雪网络中。本章应用DA模块来提高网络在现实世界中的泛化能力。(3)在雪天视频数据集 SnowScape 数据集上进行了实验分析。实验结果表明,RVDNet不仅在合成数据上,而且在现实世界中都具有优越的去雪性能。
2025-04-01 17:05:07
774
原创 利用thop计算模型的参数量params和计算量flops
有多个输入时,只需要将input(output,)更改为input(output1,output2,)即可。利用thop计算模型的参数量params和计算量flops。
2023-12-11 17:56:34
327
原创 解决:CUDA out of memory: Tried to allocate 12.11 GiB......
解决:运行推理前:export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:32。
2023-07-03 13:32:55
1280
1
原创 Python加速循环的若干个办法
Python加速循环的若干个办法: (1)使用Python封装的函数,而不是自编函数; (2)使用Numpy函数,因为Numpy底层代码语言是C++; 上面两个效果仍不好,尝试下面方法中的一种或多种: (3)将需要处理的图像分为若干个部分,使用multiprocessing多进程并行化,充分利用CPU; (4)采用numba(@njit)加速循环部分(但是与multiprocessing似乎不兼容); (5)采用Cython,修改少量代码,...
2022-05-31 15:07:51
3049
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人