Skip to content
This repository has been archived by the owner on Aug 28, 2024. It is now read-only.

results of objection dection on Android are very poor,but on PC they are very good #325

Open
julien0220 opened this issue Jun 26, 2023 · 1 comment

Comments

@julien0220
Copy link

julien0220 commented Jun 26, 2023

Can detect the correct class, but the probability is reduced

runing on pc, it gets 83%
image

running on android, it gets 55%
Screenshot_20230626-102515_YOLOv5

@NeighborhoodCoding
Copy link

There are slight difference from mobile devices,
but try this:

seed=2023
random.seed(seed)
numpy.random.seed(seed)
torch.manual_seed(seed)
if torch.cuda.is_available():
    torch.cuda.manual_seed(seed)
    torch.cuda.manual_seed_all(seed)  # all gpus
    torch.backends.cudnn.benchmark = False
    torch.backends.cudnn.deterministic = True
torch.use_deterministic_algorithms(True)
torch.set_default_tensor_type(torch.float32)

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants