fscw's Introduction

FSCW

This repository contains the code (in PyTorch) for the paper: See you somewhere in the ocean: Few-shot domain adaptive underwater object detection

The code of this repository is based on: https://github.com/ultralytics/yolov5

Our demo is based on :https://www.youtube.com/watch?v=PfEz7yE4m0U

The left side of the demo video is our proposed method, and the right side is our baseline (yolov5)

Dependencies

Python 3.7
PyTorch >=1.7
CUDA11.0 and cuDNN
numpy
tqdm
tensorboardX

Dataset

urpc 链接：https://pan.baidu.com/s/1QcTsP62KQcEiB831cGimRw

  提取码：7eym

aquarium https://universe.roboflow.com/data-science-day-dry-run/aquarium-6cfzm/dataset/1

Sourcedomaintrain

python train.py --data crossdomain.yaml --epochs 300 --weights yolov5x.pt --cfg yolov5x_FCM.yaml

Targetdomaintrain

python trainfreeze.py --data aquam.yaml --epochs 300 --weights your_source_trained_weight_path --cfg yolov5x_FCM.yaml

fscw's People

Contributors

Stargazers

Watchers

fscw's Issues

URPC Dataset Year

Which (year) URPC dataset is this based upon / is provided?

I also couldn't find the paper referenced, is it not released yet ?

training error！

hello，when i run "python train.py --data crossdomain.yaml --epochs 300 --weights yolov5x.pt --cfg yolov5x_FCM.yaml",i got an error.
the error

where can i get these .txt files,hope your response

Training Error!

Hello,

When I run the code for the training, I get an error. The error:

Thank you in advance!

hello when i run " python train.py --data crossdomain.yaml --epochs 300 --weights yolov5x.pt --cfg yolov5x_FCM.yaml"
i got an error :dataset not found

how can i change the local datasets urpc,hope tp receive your response.

Recommend Projects